diff --git a/script5_1.py b/script5_1.py index 2b167de..39884e0 100644 --- a/script5_1.py +++ b/script5_1.py @@ -13,9 +13,9 @@ data = pd.read_csv('data.csv') # Prepare the data X = data[['movie title', 'User Rating', 'Director', 'Top 5 Casts', 'Writer']] -y = data['Rating'].apply(lambda x: float(x) if isinstance(x, (int, float)) else np.nan) # Convert 'Rating' column to float data type +#y = data['Rating'].apply(lambda x: float(x) if isinstance(x, (int, float)) else np.nan) # Convert 'Rating' column to float data type #y = pd.Series(data['Rating'], dtype=float) -# y = data['Rating'].values.astype(float) +y = data['Rating'].values.astype('float64') print("Data type of 'Rating' column:", y.dtype)