mirror of
https://github.com/aladdinpersson/Machine-Learning-Collection.git
synced 2026-04-10 12:33:44 +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": {},
|
"metadata": {},
|
||||||
"outputs": [],
|
"outputs": [],
|
||||||
"source": [
|
"source": [
|
||||||
"import os \n",
|
"import os\n",
|
||||||
"\n",
|
"\n",
|
||||||
"if os.path.exists('movielens_small.zip'):\n",
|
"if os.path.exists('movielens_small.zip'):\n",
|
||||||
" !wget https://files.grouplens.org/datasets/movielens/ml-latest-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