From e61e90da0208f165b93a985a0a6e4754902dd688 Mon Sep 17 00:00:00 2001 From: Dominik Strzako Date: Fri, 14 May 2021 20:15:24 +0200 Subject: [PATCH] fix --- Zadanie_06_training.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/Zadanie_06_training.py b/Zadanie_06_training.py index c1aa885..1e2c8b8 100644 --- a/Zadanie_06_training.py +++ b/Zadanie_06_training.py @@ -21,14 +21,14 @@ x_train, x_test, y_train, y_test = train_test_split(x,y , test_size=0.2,train_si def regression_model(): model = Sequential() - model.add(Dense(32,activation = "relu", input_shape = (x_train.shape[1],))) - model.add(Dense(64,activation = "relu")) + model.add(Dense(16,activation = "relu", input_shape = (x_train.shape[1],))) + model.add(Dense(32,activation = "relu")) model.add(Dense(1,activation = "relu")) model.compile(optimizer = "adam", loss = "mean_squared_error") return model model = regression_model() -model.fit(x_train, y_train, epochs = 600, verbose = 1) +model.fit(x_train, y_train, epochs = 40, verbose = 1) model.save('wine_model') \ No newline at end of file