fix naming

javaThonc · you-n-g · commit 2b6d16feb128 · 2021-01-22T19:16:57.000+08:00
diff --git a/examples/benchmarks/TabNet/pretrain/best.model b/examples/benchmarks/TabNet/pretrain/best.model
diff --git a/examples/benchmarks/TabNet/workflow_config_TabNet_Alpha158.yaml b/examples/benchmarks/TabNet/workflow_config_TabNet_Alpha158.yaml
@@ -41,7 +41,7 @@ port_analysis_config: &port_analysis_config
         min_cost: 5
 task:
     model:
-        class: TabNet_Model
+        class: TabnetModel
         module_path: qlib.contrib.model.pytorch_tabnet
         kwargs:
             pretrain: True
diff --git a/qlib/contrib/model/pytorch_tabnet.py b/qlib/contrib/model/pytorch_tabnet.py
@@ -28,7 +28,7 @@
 from ...data.dataset.handler import DataHandlerLP
 
 
-class TabNet_Model(Model):
+class TabnetModel(Model):
     def __init__(
         self,
         d_feat=158,
