fix naming

microsoft · Jan 22, 2021 · 2b6d16f · 2b6d16f
1 parent 0a86a6f
commit 2b6d16f
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
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,