Update example_distributional.py

Adapt comments
basf · May 23, 2024 · 0cfe3b5 · 0cfe3b5
1 parent 863ff48
commit 0cfe3b5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/example_distributional.py b/example_distributional.py
@@ -7,7 +7,7 @@
 # Set random seed for reproducibility
 np.random.seed(0)
 
-# Number of samples
+# Number of samples and features
 n_samples = 1000
 n_features = 5
 
@@ -18,7 +18,7 @@
 # Generate target variable
 y = np.dot(X, coefficients) + np.random.randn(n_samples)
 
-# Create a DataFrame to store the data
+# Create a DataFrame to store the generated data
 data = pd.DataFrame(X, columns=[f"feature_{i}" for i in range(n_features)])
 data["target"] = y