README

Limmen · May 19, 2021 · 6d4412f · 6d4412f
1 parent 98ff743
commit 6d4412f
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -51,6 +51,16 @@ env_name = "optimal-intrusion-response-v1"
 env = gym.make(env_name)
 ```
 
+## Infrastructure
+
+<p align="center">
+<img src="docs/env.png" width="600">
+</p>
+
+## Traces
+
+Alert/login traces from the emulated infrastructure are available in ([./traces](./traces)).
+
 ## Publications
 
 TODO

diff --git a/docs/env.png b/docs/env.png
diff --git a/examples/v1/test_dynamics_model.py b/examples/v1/test_dynamics_model.py
@@ -10,14 +10,15 @@ def test_model() -> None:
 
     :return: None
     """
-    save_dynamics_model_dir = "/Users/kimham/workspace/gym-optimal-intrusion-response/examples/v1/"
+    # save_dynamics_model_dir = "/Users/kimham/workspace/gym-optimal-intrusion-response/examples/v1/"
+    save_dynamics_model_dir = "/Users/kimham/workspace/gym-optimal-intrusion-response/traces/"
 
     defender_dynamics_model = DefenderDynamicsModel()
     new_model = DefenderDynamicsModel()
     if save_dynamics_model_dir is not None:
         print("loading dynamics model")
-        defender_dynamics_model.read_model(save_dynamics_model_dir, model_name="new_defender_dynamics_model.json")
-        new_model.read_model(save_dynamics_model_dir, model_name="new_defender_dynamics_model.json")
+        defender_dynamics_model.read_model(save_dynamics_model_dir, model_name="traces.json")
+        new_model.read_model(save_dynamics_model_dir, model_name="traces.json")
         print("model loaded")
 
     print("normalizing model counts")

diff --git a/traces/traces.json b/traces/traces.json