CBC-covid / requirements.txt
AleksanderObuchowski's picture
change tabnet version
f93cffc
raw
history blame
472 Bytes
pandas #==1.1.5
numpy
matplotlib
seaborn
scikit-learn
xgboost
catboost
hyperopt
torch #==1.7.1+cu101
torchvision #==0.8.2+cu101
# pytorch-lightning #==1.3.6
pytorch-tabnet ==4.0
pytorch_tabular #==0.7.0
imblearn
streamlit
streamlit-lottie
hydralit_components
streamlit-modal
streamlit-echarts
# torchmetrics #==0.5.0
# tab-transformer-pytorch
# pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html