liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-16 17:12:20 +08:00
Code Activity

fix ddgda run all bug & pylint (#1031)

Browse Source
This commit is contained in:
you-n-g
2022-04-03 20:43:02 +08:00
committed by GitHub
parent fe155703b0
commit 6edd0bf298
1 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
examples/benchmarks_dynamic/DDG-DA/workflow.py
View File

@@ -9,13 +9,10 @@ from qlib.data.dataset.handler import DataHandlerLP
import pandas as pd import pandas as pd
import fire import fire
import sys import sys
from tqdm.auto import tqdm
import yaml
import pickle import pickle
from qlib import auto_init from qlib import auto_init
from qlib.model.trainer import TrainerR, task_train from qlib.model.trainer import TrainerR
from qlib.utils import init_instance_by_config from qlib.utils import init_instance_by_config
from qlib.workflow.task.gen import RollingGen, task_generator
from qlib.workflow import R from qlib.workflow import R
from qlib.tests.data import GetData from qlib.tests.data import GetData
@@ -47,9 +44,10 @@ class DDGDA:
rb = RollingBenchmark(model_type="gbdt") rb = RollingBenchmark(model_type="gbdt")
task = rb.basic_task() task = rb.basic_task()
model = init_instance_by_config(task["model"]) with R.start(experiment_name="feature_importance"):
dataset = init_instance_by_config(task["dataset"]) model = init_instance_by_config(task["model"])
model.fit(dataset) dataset = init_instance_by_config(task["dataset"])
model.fit(dataset)
fi = model.get_feature_importance() fi = model.get_feature_importance()