adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

make BaseClassifierModel. Add predict_proba to lightgbm

Browse Source
This commit is contained in:
robcaulk
2022-08-13 20:07:31 +02:00
parent 31be707cc8
commit 58de20af0f
4 changed files with 108 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
freqtrade/freqai/prediction_models/LightGBMClassifier.py
View File

@@ -3,13 +3,13 @@ from typing import Any, Dict
from lightgbm import LGBMClassifier
from freqtrade.freqai.prediction_models.BaseRegressionModel import BaseRegressionModel
from freqtrade.freqai.prediction_models.BaseClassifierModel import BaseClassifierModel
logger = logging.getLogger(__name__)
class LightGBMClassifier(BaseRegressionModel):
class LightGBMClassifier(BaseClassifierModel):
"""
User created prediction model. The class needs to override three necessary
functions, predict(), train(), fit(). The class inherits ModelHandler which