diff --git a/script5_1.py b/script5_1.py
index f947e16..1f561f5 100644
--- 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