This commit is contained in:
Yevhenii Poliakov 2023-05-14 21:24:33 +02:00
parent 8fbf51e92e
commit 3163212f32

View File

@ -15,6 +15,9 @@ data = pd.read_csv('data.csv')
X = data[['movie title', 'User Rating', 'Director', 'Top 5 Casts', 'Writer']]
y = data['Rating'].astype(float)
print("Data type of 'Rating' column:", y.dtype)
# Preprocess the data
# Convert the categorical columns into numerical representations
mlb_genres = MultiLabelBinarizer()