digital-forensics-labs/Machine-Learning-Collection
1
0
Fork 0
mirror of https://github.com/aladdinpersson/Machine-Learning-Collection.git synced 2026-04-10 12:33:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

updated readme for docker

Browse Source
This commit is contained in:
Aladdin Persson
2023-12-08 15:59:55 +01:00
parent 4455010336
commit 1982b04dc1
2 changed files with 62 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
requirements.txt
View File

@@ -1,3 +1,5 @@
# requirements.txt file with basic libraries to install for a machine learning workflow
numpy
pandas
scikit-learn
@@ -5,24 +7,35 @@ matplotlib
seaborn
scipy
# Deep Learning
# deep learning
torchvision
torchaudio
transformers
tensorboard
lightning
# For gradient boosting machines (GBMs)
# all you need is xgboost
xgboost
lightgbm
# For working with text data
# nlp libraries
nltk
spacy
# For image processing tasks
# image processing
opencv-python-headless
Pillow
# Jupyter Notebook (remove if you do not use it within the container)
# data loading
pyarrow
# model optimization/experiment tracking
tensorboard
wandb
mlflow
# utilities
tqdm
# notebooks
jupyter
ipywidgets
ipywidgets