mirror of
https://github.com/aladdinpersson/Machine-Learning-Collection.git
synced 2026-02-20 13:50:41 +00:00
added Dockerfile setup
This commit is contained in:
9
.dockerignore
Normal file
9
.dockerignore
Normal file
@@ -0,0 +1,9 @@
|
||||
__pycache__
|
||||
.git
|
||||
.gitignore
|
||||
.vscode
|
||||
.idea
|
||||
*.pyc
|
||||
*~
|
||||
data/
|
||||
secrets.env
|
||||
4
Dockerfile
Normal file
4
Dockerfile
Normal file
@@ -0,0 +1,4 @@
|
||||
FROM pytorch/pytorch:latest
|
||||
WORKDIR /code
|
||||
RUN pip install --no-cache-dir -r requirements.txt
|
||||
EXPOSE 8888
|
||||
@@ -13,7 +13,7 @@
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"import os \n",
|
||||
"import os\n",
|
||||
"\n",
|
||||
"if os.path.exists('movielens_small.zip'):\n",
|
||||
" !wget https://files.grouplens.org/datasets/movielens/ml-latest-small.zip \n",
|
||||
|
||||
@@ -0,0 +1,6 @@
|
||||
# TF-IDF from scratch
|
||||
|
||||
import numpy as np
|
||||
import pandas as pd
|
||||
|
||||
|
||||
28
requirements.txt
Normal file
28
requirements.txt
Normal file
@@ -0,0 +1,28 @@
|
||||
numpy
|
||||
pandas
|
||||
scikit-learn
|
||||
matplotlib
|
||||
seaborn
|
||||
scipy
|
||||
|
||||
# Deep Learning
|
||||
torchvision
|
||||
torchaudio
|
||||
transformers
|
||||
tensorboard
|
||||
|
||||
# For gradient boosting machines (GBMs)
|
||||
xgboost
|
||||
lightgbm
|
||||
|
||||
# For working with text data
|
||||
nltk
|
||||
spacy
|
||||
|
||||
# For image processing tasks
|
||||
opencv-python-headless
|
||||
Pillow
|
||||
|
||||
# Jupyter Notebook (remove if you do not use it within the container)
|
||||
jupyter
|
||||
ipywidgets
|
||||
Reference in New Issue
Block a user