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

update version number & model license

Browse Source
This commit is contained in:
Young
2020-11-28 08:17:18 +00:00
parent 05afeb820a
commit 3f47a282cc
7 changed files with 25 additions and 163 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
qlib/model/base.py
View File

@@ -56,6 +56,23 @@ class ModelFT(Model):
def finetune(self, dataset: Dataset):
"""finetune model based given dataset
A typical use case of finetuning model with qlib.workflow.R
.. code-block:: python
# start exp to train init model
with R.start(experiment_name="init models"):
model.fit(dataset)
R.save_objects(init_model=model)
rid = R.get_recorder().id
# Finetune model based on previous trained model
with R.start(experiment_name="finetune model"):
recorder = R.get_recorder(rid, experiment_name="init models")
model = recorder.load_object("init_model")
model.finetune(dataset, num_boost_round=10)
Parameters
----------
dataset : Dataset