liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-14 00:06:58 +08:00
Code Activity

fix naming

Browse Source
This commit is contained in:
Alex Wang
2021-01-22 18:20:52 +09:00
committed by you-n-g
parent 0a86a6f392
commit 2b6d16feb1
3 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

BIN
examples/benchmarks/TabNet/pretrain/best.model
View File

Binary file not shown.

2
examples/benchmarks/TabNet/workflow_config_TabNet_Alpha158.yaml
View File

@@ -41,7 +41,7 @@ port_analysis_config: &port_analysis_config
min_cost: 5 min_cost: 5
task: task:
model: model:
class: TabNet_Model class: TabnetModel
module_path: qlib.contrib.model.pytorch_tabnet module_path: qlib.contrib.model.pytorch_tabnet
kwargs: kwargs:
pretrain: True pretrain: True

2
qlib/contrib/model/pytorch_tabnet.py
View File

@@ -28,7 +28,7 @@ from ...data.dataset import DatasetH
from ...data.dataset.handler import DataHandlerLP from ...data.dataset.handler import DataHandlerLP
class TabNet_Model(Model): class TabnetModel(Model):
def __init__( def __init__(
self, self,
d_feat=158, d_feat=158,