Merge pull request #10 from vluzko/nri

Added scripts to gitignore
vluzko · Mar 26, 2021 · 92ee221 · 92ee221
2 parents dcdf50c + 9007419
commit 92ee221
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -7,3 +7,4 @@ __pycache__/
 data/
 plots/
 runs/
+scripts/
diff --git a/causal_rl/nri.py b/causal_rl/nri.py
@@ -0,0 +1,13 @@
+"""A model based on Neural Relational Inference."""
+
+
+class NRIModel:
+    pass
+
+
+class Encoder:
+    pass
+
+
+class Decoder:
+    pass
-Original file line number
+Diff line change
@@ Expand Up / @@ -7,3 +7,4 @@ __pycache__/ @@
     data/
     plots/
     runs/
+    scripts/