s430705/ium_430705
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Zaktualizuj 'lab06_training.py'
Some checks failed
s430705-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Michał Zaręba 2021-05-10 15:41:38 +02:00
parent f9669b1bc6
commit b0e1ae3e32
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lab06_training.py
View File

@ -19,7 +19,7 @@ movies_data.dropna(inplace=True)
X = movies_data.drop("rating", axis=1) X = movies_data.drop("rating", axis=1)
Y = movies_data["rating"] Y = movies_data["rating"]
print(X, Y.values)
# Split set to train/test 8:2 ratio # Split set to train/test 8:2 ratio
X_train, X_test, Y_train, Y_test = train_test_split( X_train, X_test, Y_train, Y_test = train_test_split(
X, Y, test_size=0.2, random_state=42 X, Y, test_size=0.2, random_state=42