add pytorch mlp models to test_start_backtesting

2023-03-21 16:20:35 +02:00
parent 83a7d888bc
commit 02bccd0097
2 changed files with 27 additions and 27 deletions
@@ -83,6 +83,22 @@ def make_rl_config(conf):
    return conf


+def mock_pytorch_mlp_model_training_parameters(conf):
+    return {
+            "learning_rate": 3e-4,
+            "trainer_kwargs": {
+                "max_iters": 1,
+                "batch_size": 64,
+                "max_n_eval_batches": 1,
+            },
+            "model_kwargs": {
+                "hidden_dim": 32,
+                "dropout_percent": 0.2,
+                "n_layer": 1,
+            }
+        }
+
+
 def get_patched_data_kitchen(mocker, freqaiconf):
    dk = FreqaiDataKitchen(freqaiconf)
    return dk