Skip to content

Commit d618102

Browse files
author
Sergey Feldman
committed
fix reqs
1 parent 1511fb8 commit d618102

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

03_autogluon.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
from utils import load_data
1010

1111

12-
N_JOBS = 4 * 4 * 9
1312
SEC = 60
1413

1514

requirements.in

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
numpy
22
scipy
33
pandas
4-
scikit-learn>=0.24
5-
lightgbm>=3.1.1
64
seaborn
7-
mxnet_cu101 # or mxnet if don't have a gpu
8-
autogluon
5+
mxnet_cu102 # insert your own cuda number OR 'mxnet' if don't have a gpu
6+
autogluon
7+
scikit-learn>=0.24
8+
lightgbm>=3.1.1

0 commit comments

Comments
 (0)