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

Add install variant for freqai

Browse Source
This commit is contained in:
Matthias 2022-07-31 15:10:01 +02:00
parent 61693f6c8b
commit cbb05354a8
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
requirements-freqai.txt
View File

@ -3,7 +3,6 @@
# Required for freqai # Required for freqai
scikit-learn==1.1.1 scikit-learn==1.1.1
scikit-optimize==0.9.0
joblib==1.1.0 joblib==1.1.0
catboost==1.0.4 catboost==1.0.4
lightgbm==3.3.2 lightgbm==3.3.2

10
setup.py
View File

@ -12,6 +12,13 @@ hyperopt = [
'progressbar2', 'progressbar2',
] ]
freqai = [
'scikit-learn',
'joblib',
'catboost',
'lightgbm',
]
develop = [ develop = [
'coveralls', 'coveralls',
'flake8', 'flake8',
@ -31,7 +38,7 @@ jupyter = [
'nbconvert', 'nbconvert',
] ]
all_extra = plot + develop + jupyter + hyperopt all_extra = plot + develop + jupyter + hyperopt + freqai
setup( setup(
tests_require=[ tests_require=[
@ -79,6 +86,7 @@ setup(
'plot': plot, 'plot': plot,
'jupyter': jupyter, 'jupyter': jupyter,
'hyperopt': hyperopt, 'hyperopt': hyperopt,
'freqai': freqai,
'all': all_extra, 'all': all_extra,
}, },
) )