update script5

This commit is contained in:
Yevhenii Poliakov 2023-05-14 19:59:54 +02:00
parent bdb8004c8d
commit 8259772b18

View File

@ -9,6 +9,8 @@ from tensorflow.keras.optimizers import Adam
# Load the dataset from the CSV file # Load the dataset from the CSV file
data = pd.read_csv('data.csv') data = pd.read_csv('data.csv')
data = data.drop(index=5059)
# Prepare the data # Prepare the data
X = data[['movie title', 'User Rating', 'Genres', 'Plot Keyword', 'Director', 'Top 5 Casts', 'Writer', 'Year']] X = data[['movie title', 'User Rating', 'Genres', 'Plot Keyword', 'Director', 'Top 5 Casts', 'Writer', 'Year']]
y = data['Rating'] y = data['Rating']