1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-09 22:10:56 +08:00

Update xgb parameters.

This commit is contained in:
lwwang1995
2020-11-25 22:05:36 +08:00
parent 27f19c1f1b
commit f84cb64332
2 changed files with 10 additions and 15 deletions

View File

@@ -22,10 +22,8 @@ from ...data.dataset.handler import DataHandlerLP
class XGBModel(Model):
"""XGBModel Model"""
def __init__(self, obj="mse", **kwargs):
if obj not in {"mse", "binary"}:
raise NotImplementedError
self._params = {"obj": obj}
def __init__(self, **kwargs):
self._params = {}
self._params.update(kwargs)
self.model = None