mirror of
https://github.com/aladdinpersson/Machine-Learning-Collection.git
synced 2026-02-21 11:18:01 +00:00
42 lines
467 B
Plaintext
42 lines
467 B
Plaintext
# requirements.txt file with basic libraries to install for a machine learning workflow
|
|
|
|
numpy
|
|
pandas
|
|
scikit-learn
|
|
matplotlib
|
|
seaborn
|
|
scipy
|
|
|
|
# deep learning
|
|
torchvision
|
|
torchaudio
|
|
transformers
|
|
lightning
|
|
|
|
# all you need is xgboost
|
|
xgboost
|
|
lightgbm
|
|
|
|
# nlp libraries
|
|
nltk
|
|
spacy
|
|
|
|
# image processing
|
|
opencv-python-headless
|
|
Pillow
|
|
|
|
# data loading
|
|
pyarrow
|
|
|
|
# model optimization/experiment tracking
|
|
tensorboard
|
|
wandb
|
|
mlflow
|
|
|
|
# utilities
|
|
tqdm
|
|
|
|
# notebooks
|
|
jupyter
|
|
ipywidgets
|