This commit is contained in:
Yevhenii Poliakov 2023-05-14 21:28:44 +02:00
parent e332f40a95
commit da57700f7b

View File

@ -13,7 +13,8 @@ data = pd.read_csv('data.csv')
# Prepare the data
X = data[['movie title', 'User Rating', 'Director', 'Top 5 Casts', 'Writer']]
y = data['Rating'].values.astype(float)
y = pd.Series(data['Rating'], dtype=float)
# y = data['Rating'].values.astype(float)
print("Data type of 'Rating' column:", y.dtype)