Fix backward compatibility of NNOptimiserParams

opcode81 · Jul 29, 2023 · c9f73af · c9f73af
1 parent f08dccd
commit c9f73af
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/sensai/torch/torch_opt.py b/src/sensai/torch/torch_opt.py
@@ -538,7 +538,15 @@ def get_validation_metric_name(self):
 class NNOptimiserParams(ToStringMixin):
     REMOVED_PARAMS = {"cuda"}
     RENAMED_PARAMS = {
-        "optimiserClip": "shrinkageClip"
+        "optimiserClip": "optimiser_clip",
+        "lossEvaluator": "loss_evaluator",
+        "optimiserLR": "optimiser_lr",
+        "earlyStoppingEpochs": "early_stopping_epochs",
+        "batchSize": "batch_size",
+        "trainFraction": "train_fraction",
+        "scaledOutputs": "scaled_outputs",
+        "useShrinkage": "use_shrinkage",
+        "shrinkageClip": "shrinkage_clip",
     }
 
     def __init__(self,