Update app.py
Browse files
app.py
CHANGED
|
@@ -63,7 +63,7 @@ for col in iris['feature_names']:
|
|
| 63 |
|
| 64 |
from sklearn.model_selection import train_test_split
|
| 65 |
|
| 66 |
-
X_train, X_test, y_train, y_test = train_test_split(X,
|
| 67 |
|
| 68 |
"""### ML
|
| 69 |
|
|
|
|
| 63 |
|
| 64 |
from sklearn.model_selection import train_test_split
|
| 65 |
|
| 66 |
+
X_train, X_test, y_train, y_test = train_test_split(X, df['species'], test_size=0.2, random_state=42)
|
| 67 |
|
| 68 |
"""### ML
|
| 69 |
|