s151636/ium_151636
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

upd 5

Browse Source
This commit is contained in:
Yevhenii Poliakov 2023-05-14 20:27:17 +02:00
parent 27860c181b
commit e1110926d2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
script5.py
View File

@ -26,7 +26,7 @@ X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_
# Create the neural network model
model = Sequential()
model.add(Embedding(input_dim=len(mlb.classes_), output_dim=10, input_length=X.shape[24402, 39]))
model.add(Embedding(input_dim=len(mlb.classes_), output_dim=10, input_length=X.shape[1]))
model.add(Flatten())
model.add(Dense(32, activation='relu'))
model.add(Dense(1))