From 9947cf587fccce32cdfbe40e0aa3cd9516c4b899 Mon Sep 17 00:00:00 2001 From: s434695 Date: Mon, 17 May 2021 12:32:14 +0200 Subject: [PATCH] fix --- train.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/train.py b/train.py index d15a2bb..67d14d7 100755 --- a/train.py +++ b/train.py @@ -43,8 +43,10 @@ def prepare_model(epoch_param, batch_param, _run): X_test = vgsales_test.drop(['Rank','Name','Platform','Year','Genre','Publisher'],axis = 1) y_test = vgsales_test[['Nintendo']] + print(X_train.shape[1]) + model = Sequential() - model.add(Dense(9, input_dim = X_train, kernel_initializer='normal', activation='relu')) + model.add(Dense(9, input_dim = X_train.shape[1], kernel_initializer='normal', activation='relu')) model.add(Dense(1,kernel_initializer='normal', activation='sigmoid')) early_stop = EarlyStopping(monitor="val_loss", mode="min", verbose=1, patience=10)