This commit is contained in:
Yevhenii Poliakov 2023-05-14 21:21:13 +02:00
parent f9d635c67d
commit 8fbf51e92e

View File

@ -13,7 +13,7 @@ data = pd.read_csv('data.csv')
# Prepare the data # Prepare the data
X = data[['movie title', 'User Rating', 'Director', 'Top 5 Casts', 'Writer']] X = data[['movie title', 'User Rating', 'Director', 'Top 5 Casts', 'Writer']]
y = data['Rating'] y = data['Rating'].astype(float)
# Preprocess the data # Preprocess the data
# Convert the categorical columns into numerical representations # Convert the categorical columns into numerical representations