add finetune example & fix serial bug

2026-07-12 23:36:54 +08:00 · 2020-11-16 13:11:39 +00:00
parent 3e04ded750
commit 90d41e4022
5 changed files with 203 additions and 37 deletions
--- a/qlib/model/base.py
+++ b/qlib/model/base.py
@@ -45,3 +45,18 @@ class Model(BaseModel):
            dataset will generate the processed dataset from model training
        """
        raise NotImplementedError()
+
+
+class ModelFT(Model):
+    '''Model (F)ine(t)unable'''
+
+    @abc.abstractmethod
+    def finetune(self, dataset: Dataset):
+        """finetune model based given dataset
+
+        Parameters
+        ----------
+        dataset : Dataset
+            dataset will generate the processed dataset from model training
+        """
+        raise NotImplementedError()