First stable version for track particle

fialhocoelho · Feb 6, 2020 · d8a94d0 · d8a94d0
commit d8a94d0
Show file tree

Hide file tree

Showing 37 changed files with 103,795 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,11 @@
+__pycache__/
+*.py[cod]
+*$py.class
+*.pyc
+
+results/
+outputs/
+logs/
+
+.ipynb_checkpoints
+*/.ipynb_checkpoints/*
diff --git a/config-paralel.json b/config-paralel.json
@@ -0,0 +1,46 @@
+{
+    "data": {
+        "filename": "/data/track-ml/phi025-025_eta025-025/phi025-025_eta025-025_filtered.csv",
+        "train_split": 0.70,
+        "normalise": true
+    },
+    "training": {
+        "epochs": 20,
+        "batch_size": 5
+    },
+
+    "model": {
+        "name": "lstm-paralel",
+        "loss": "mse",
+        "optimizer": "RMSprop",
+        "save_dir": "results/",
+        "layers": [
+            {
+                "type": "lstm-paralel",
+                "neurons": 400,
+                "input_timesteps": 4,
+                "input_features": 3,
+                "return_seq": false
+            },
+            {
+                "type": "dense",
+                "neurons": 400,
+                "activation": "relu"
+            },             
+            {
+                "type": "dropout",
+                "rate": 0.2
+            },
+            {
+                "type": "dense",
+                "neurons": 20,
+                "activation": "relu"
+            },              
+            {
+                "type": "dense",
+                "neurons": 3,
+                "activation": "linear"
+            }
+        ]
+    }
+}
diff --git a/config.json b/config.json
@@ -0,0 +1,43 @@
+{
+    "data": {
+        "filename": "./dataset/event_2000_2020_train.csv",
+        "train_split": 0.80,
+        "normalise": true
+    },
+    "training": {
+        "epochs": 30,
+        "batch_size": 20
+    },
+
+    "model": {
+        "name": "lstm",
+        "loss": "mse",
+        "optimizer": "adam",
+        "save_dir": "results/",
+        "layers": [
+            {
+                "type": "lstm",
+                "neurons": 20,
+                "input_timesteps": 4,
+                "input_features": 3,
+                "return_seq": true
+            },
+            {
+                "type": "dropout",
+                "rate": 0.2
+            },
+            {
+                "type": "lstm",
+                "neurons": 20,
+                "input_timesteps": 4,
+                "input_features": 3,
+                "return_seq": true
+            },                          
+            {
+                "type": "dense",
+                "neurons": 3,
+                "activation": "linear"
+            }
+        ]
+    }
+}