upd51

2023-05-14 21:21:13 +02:00 · 2023-05-14 21:21:13 +02:00 · 8fbf51e92e
commit 8fbf51e92e
parent f9d635c67d
1 changed files with 1 additions and 1 deletions
--- a/script5_1.py
+++ b/script5_1.py
@ -13,7 +13,7 @@ data = pd.read_csv('data.csv')

 # Prepare the data
 X = data[['movie title', 'User Rating', 'Director', 'Top 5 Casts', 'Writer']]
-y = data['Rating']
+y = data['Rating'].astype(float)

 # Preprocess the data
 # Convert the categorical columns into numerical representations