This commit is contained in:
Yevhenii Poliakov 2023-05-14 20:28:28 +02:00
parent e1110926d2
commit 7e7ee5b0c9

View File

@ -18,7 +18,7 @@ y = data['Rating']
# Convert the categorical columns into numerical representations
mlb = MultiLabelBinarizer()
X['Generes'] = mlb.fit_transform(X['Generes'])
X['Plot Keyword'] = mlb.fit_transform(X['Plot Kyeword'])
X['Plot Kyeword'] = mlb.fit_transform(X['Plot Kyeword'])
X['Top 5 Casts'] = mlb.fit_transform(X['Top 5 Casts'])
# Split the data into training and testing sets