revisions

.gitignore

@@ -162,4 +162,5 @@ checkpoints/
 results/
 result_long_term_forecast.txt
 result_anomaly_detection.txt
-scripts/augmentation/
+scripts/augmentation/
+run_anylearn.py

exp/exp_classification.py

@@ -132,7 +132,7 @@ def train(self, setting):
             print(
                 "Epoch: {0}, Steps: {1} | Train Loss: {2:.3f} Vali Loss: {3:.3f} Vali Acc: {4:.3f} Test Loss: {5:.3f} Test Acc: {6:.3f}"
                 .format(epoch + 1, train_steps, train_loss, vali_loss, val_accuracy, test_loss, test_accuracy))
-
+            early_stopping(-val_accuracy, self.model, path)
             if early_stopping.early_stop:
                 print("Early stopping")
                 break

requirements.txt

@@ -10,4 +10,4 @@ sktime
 sympy
 torch
 tqdm
-# mamba_ssm
+mamba_ssm