s151636/ium_151636
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

upd 5

Browse Source
This commit is contained in:
Yevhenii Poliakov 2023-05-14 20:34:27 +02:00
parent 7e7ee5b0c9
commit 830d0030dd
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
script5.py
View File

@ -18,8 +18,8 @@ y = data['Rating']
# Convert the categorical columns into numerical representations
mlb = MultiLabelBinarizer()
X['Generes'] = mlb.fit_transform(X['Generes'])
X['Plot Kyeword'] = mlb.fit_transform(X['Plot Kyeword'])
X['Top 5 Casts'] = mlb.fit_transform(X['Top 5 Casts'])
X['Plot Keyword'] = mlb.fit_transform(X['Plot Kyeword'])
X['Top 5 Casts'] = mlb.fit_transform(X['Top 5 Casts'].astype(str))
# Split the data into training and testing sets
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)