update script5

This commit is contained in:
Yevhenii Poliakov 2023-05-14 20:02:54 +02:00
parent 8259772b18
commit 5f2f9c152f

View File

@ -7,9 +7,8 @@ from tensorflow.keras.layers import Dense, Embedding, Flatten
from tensorflow.keras.optimizers import Adam 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', on_bad_lines='skip')
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']]