requirements.txt

# =====================================================
# PYTHON VERSION
# Recommended: Python 3.10 or 3.11
# =====================================================


# =====================================================
# CORE DATA SCIENCE
# =====================================================

numpy==1.26.4
pandas==2.2.2
scikit-learn==1.5.1
scipy==1.13.1


# =====================================================
# VISUALIZATION
# =====================================================

matplotlib==3.9.0
seaborn==0.13.2
plotly>=5.20.0


# =====================================================
# NLP PREPROCESSING
# =====================================================

nltk==3.8.1
wordcloud==1.9.3
regex==2024.5.15


# =====================================================
# MACHINE LEARNING MODELS
# =====================================================

xgboost==2.1.1
lightgbm==4.5.0


# =====================================================
# DEEP LEARNING (TensorFlow/Keras)
# =====================================================

tensorflow==2.16.1
keras==3.3.3


# =====================================================
# PYTORCH
# =====================================================

torch==2.3.1
torchvision==0.18.1
torchaudio==2.3.1


# =====================================================
# TRANSFORMERS / BERT
# =====================================================

transformers==4.41.2
datasets==2.19.1
huggingface_hub==0.23.2
accelerate==0.30.1
peft==0.11.1
sentencepiece==0.2.0


# =====================================================
# HYPERPARAMETER TUNING
# =====================================================

optuna==3.6.1


# =====================================================
# STREAMLIT DEPLOYMENT
# =====================================================

streamlit==1.37.1


# =====================================================
# MODEL SAVING / UTILITIES
# =====================================================

joblib==1.4.2
pickle-mixin==1.0.2
tqdm==4.66.4


# =====================================================
# NOTEBOOK SUPPORT
# =====================================================

notebook==7.2.1
ipykernel==6.29.5
jupyterlab==4.2.4