liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-06-06 05:51:17 +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
task:
model:
class: TabNet_Model
class: TabnetModel
module_path: qlib.contrib.model.pytorch_tabnet
kwargs:
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
class TabNet_Model(Model):
class TabnetModel(Model):
def __init__(
self,
d_feat=158,