This commit is contained in:
Yevhenii Poliakov 2023-05-14 20:36:09 +02:00
parent 830d0030dd
commit d8e55bb489

View File

@ -18,7 +18,7 @@ y = data['Rating']
# Convert the categorical columns into numerical representations # Convert the categorical columns into numerical representations
mlb = MultiLabelBinarizer() mlb = MultiLabelBinarizer()
X['Generes'] = mlb.fit_transform(X['Generes']) 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'].astype(str)) X['Top 5 Casts'] = mlb.fit_transform(X['Top 5 Casts'].astype(str))
# Split the data into training and testing sets # Split the data into training and testing sets