From 22d93bb82e674089c088ee8a92ef406fa674683b Mon Sep 17 00:00:00 2001 From: AWieczarek Date: Sun, 5 May 2024 19:34:42 +0200 Subject: [PATCH] IUM_06 --- IUM_05-model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/IUM_05-model.py b/IUM_05-model.py index 89e1990..0f8a5a6 100644 --- a/IUM_05-model.py +++ b/IUM_05-model.py @@ -23,6 +23,6 @@ model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy']) -model.fit(X_train_pad, y_train, epochs=sys.argv[1], batch_size=sys.argv[2], validation_split=0.1) +model.fit(X_train_pad, y_train, epochs=int(sys.argv[1]), batch_size=int(sys.argv[2]), validation_split=0.1) model.save('beer_review_sentiment_model.h5')