Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -55,7 +55,7 @@ def create_plot(alpha, seed):
|
|
55 |
fig = plt.figure(figsize=(7, 7))
|
56 |
for i, ds in enumerate(datasets):
|
57 |
X, y = ds
|
58 |
-
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.4, random_state=
|
59 |
model.fit(X_train, y_train)
|
60 |
|
61 |
ax = fig.add_subplot(3, 2, 2*i+1)
|
|
|
55 |
fig = plt.figure(figsize=(7, 7))
|
56 |
for i, ds in enumerate(datasets):
|
57 |
X, y = ds
|
58 |
+
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.4, random_state=seed)
|
59 |
model.fit(X_train, y_train)
|
60 |
|
61 |
ax = fig.add_subplot(3, 2, 2*i+1)
|