huabdul commited on
Commit
eea52a0
1 Parent(s): cf7215e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=SEED)
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)