Test : complete test

Author: NE7K

Image/ConfMatrix/fold_1.png

@@ -39,7 +39,7 @@ def client(self) -> Client:
 @dataclass(frozen=True)
 class PathConfig:
     sentiment_data: str = 'DataSets/upstock-sentiment-data.csv' # sentiment data
-    tokenizer: str = 'SaveModel/upstock_sentiment_tokenizer.pickle'
+    tokenizer: str = 'SaveModel/upstock_sentiment_tokenizer.pkl'
     model_gz: str = 'SaveModel/upstock_sentiment_model.keras.gz'
     model_h5_gz: str = 'SaveModel/upstock_sentiment_model.h5.gz'
     model: str = 'SaveModel/upstock_sentiment_model.keras'

Image/ConfMatrix/fold_2.png

@@ -106,13 +106,13 @@ def run_train():
 
         callbacks = [
             TensorBoard(log_dir=log_dir),
-            EarlyStopping(monitor='val_loss', patience=20, restore_best_weights=True)
+            EarlyStopping(monitor='val_loss', patience=5, restore_best_weights=True)
         ]
 
         history = model.fit(
             X_train, y_train,
             validation_data=(X_val, y_val),
-            epochs=20,
+            epochs=10,
             batch_size=32,
             class_weight=class_weights,
             callbacks=callbacks,