added Dockerfile setup

This commit is contained in:
Aladdin Persson
2023-12-08 14:12:21 +01:00
parent 5280b33a5c
commit 1b310a3422
5 changed files with 48 additions and 1 deletions

9
.dockerignore Normal file
View File

@@ -0,0 +1,9 @@
__pycache__
.git
.gitignore
.vscode
.idea
*.pyc
*~
data/
secrets.env

4
Dockerfile Normal file
View File

@@ -0,0 +1,4 @@
FROM pytorch/pytorch:latest
WORKDIR /code
RUN pip install --no-cache-dir -r requirements.txt
EXPOSE 8888

View File

@@ -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",

View File

@@ -0,0 +1,6 @@
# TF-IDF from scratch
import numpy as np
import pandas as pd

28
requirements.txt Normal file
View 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