DOMMETI commited on
Commit
e395202
·
verified ·
1 Parent(s): 518e867

Update pages/1_User_Defined_DataLab.py

Browse files
Files changed (1) hide show
  1. pages/1_User_Defined_DataLab.py +2 -1
pages/1_User_Defined_DataLab.py CHANGED
@@ -17,9 +17,10 @@ st.title("🧠 Neural Network Playground - Custom Dataset")
17
  n_samples = st.slider("Number of Samples", 100, 1000, 300)
18
  noise = st.slider("Noise Level", 0.0, 1.0, 0.2)
19
  random_state = st.number_input("Random State", value=42)
 
20
 
21
  X, y = make_classification(n_samples=n_samples, n_features=2, n_redundant=0,
22
- n_informative=2, n_clusters_per_class=1,
23
  flip_y=noise, random_state=random_state, class_sep=4)
24
 
25
  df = pd.DataFrame(X, columns=["X1", "X2"])
 
17
  n_samples = st.slider("Number of Samples", 100, 1000, 300)
18
  noise = st.slider("Noise Level", 0.0, 1.0, 0.2)
19
  random_state = st.number_input("Random State", value=42)
20
+ n_clusters_per_class=st.slider("Number of Clusters",1,4,1)
21
 
22
  X, y = make_classification(n_samples=n_samples, n_features=2, n_redundant=0,
23
+ n_informative=2, n_clusters_per_class=n_clusters_per_class,
24
  flip_y=noise, random_state=random_state, class_sep=4)
25
 
26
  df = pd.DataFrame(X, columns=["X1", "X2"])