Merge pull request #124 from dice-group/docker

Add Dockerfile

Dockerfile

@@ -0,0 +1,6 @@
+FROM python:3.10
+#RUN pip install dicee==0.0.4
+RUN pip3 install "pandas>=1.5.1" "torch>=2.0.0" "polars>=0.16.14" "scikit-learn>=1.2.2" "pyarrow>=11.0.0" "pytest>=7.2.2" "gradio>=3.23.0" "psutil>=5.9.4" "pytorch-lightning==1.6.4" "pykeen==1.10.1" "zstandard>=0.21.0"
+WORKDIR /dicee
+ADD . .
+CMD ./main.py

README.md

@@ -66,6 +66,17 @@ pyreverse dicee/trainer && dot -Tpng -x classes.dot -o trainer.png && eog traine
 ```
 </details>
 
+## Docker
+To build the Docker image:
+```
+docker build -t dice-embeddings .
+```
+
+To test the Docker image:
+```
+docker run --rm -v ~/.local/share/dicee/KGs:/dicee/KGs dice-embeddings ./main.py --model AConEx --embedding_dim 16
+```
+
 # Knowledge Graph Embedding Models
 <details> <summary> Details</summary>
 

main.py

@@ -1,3 +1,4 @@
+#!/usr/bin/env python3
 import json
 from dicee.executer import Execute
 import pytorch_lightning as pl