remove dependency to catboost/lightgbm/xgboost

pull/38/head
Taiga Noumi 2020-02-15 17:13:12 +09:00
parent 7b390b976c
commit 50007e0b8b
5 changed files with 4 additions and 9 deletions

View File

@ -21,7 +21,7 @@ jobs:
run: |
python -m pip install --upgrade pip
pip install -r requirements.txt
pip install torch mlflow
pip install torch mlflow catboost lightgbm xgboost
- name: Install MeCab
run: |
sudo apt install mecab libmecab-dev mecab-ipadic-utf8

View File

@ -17,7 +17,7 @@ jobs:
run: |
python -m pip install --upgrade pip
pip install -r requirements.txt
pip install torch mlflow
pip install torch mlflow catboost lightgbm xgboost
pip install setuptools==42.0.2
pip install wheel twine
- name: Build and publish

View File

@ -47,6 +47,7 @@ def requires_lightgbm():
try:
import catboost
_has_catboost = True
# TODO check catboost version >= 0.17
except ImportError:
_has_catboost = False

View File

@ -1,6 +1,4 @@
catboost>=0.17
category_encoders
lightgbm
matplotlib
more-itertools
numpy
@ -11,4 +9,3 @@ seaborn
sklearn
tqdm
transformers
xgboost

View File

@ -28,9 +28,7 @@ setup(
license='MIT',
install_requires=[
'catboost>=0.17',
'category_encoders',
'lightgbm',
'matplotlib',
'more-itertools',
'numpy',
@ -40,8 +38,7 @@ setup(
'seaborn',
'sklearn',
'tqdm',
'transformers>=2.3.0',
'xgboost'
'transformers>=2.3.0'
],
author='nyanp',
author_email='Noumi.Taiga@gmail.com',