Successfully run training

2026-07-10 22:36:55 +08:00 · 2024-07-10 06:25:30 +00:00
parent a9fc3435ab
commit 4c057f645e
2 changed files with 52 additions and 8 deletions
--- a/tests/model/test_general_nn.py
+++ b/tests/model/test_general_nn.py
@@ -67,9 +67,16 @@ class TestNN(TestAutoData):
                    "dropout":0.,
                },
            ),
+            GeneralPTNN(
+                n_epochs=2,
+                pt_model_uri="qlib.contrib.model.pytorch_nn.Net",  # it is a MLP
+                pt_model_kwargs={
+                    "input_dim":3,
+                },
+            ),
        ]
        
-        for ds, model in zip((tsds, tbds), model_l):
+        for ds, model in reversed(list(zip((tsds, tbds), model_l))):
            model.fit(ds)  # It works
            model.predict(ds)  # It works
            break