Merge pull request #46 from Grutschus/paper-visualization

Paper visualization
Grutschus · Jan 8, 2024 · 52d2b6b · 52d2b6b
2 parents 28764b5 + c167e30
commit 52d2b6b
Show file tree

Hide file tree

Showing 9 changed files with 2,188 additions and 59 deletions.
diff --git a/.gitignore b/.gitignore
@@ -166,3 +166,4 @@ cython_debug/
 slurm-*.out
 /experiments
 /model_tests
+/diagrams
diff --git a/diagrams.dvc b/diagrams.dvc
@@ -0,0 +1,6 @@
+outs:
+- md5: 4b5c4d731c1f713db2248c1667716ec8.dir
+ size: 609116
+ nfiles: 8
+ hash: md5
+ path: diagrams
diff --git a/environment.yml b/environment.yml
@@ -16,6 +16,7 @@ dependencies:
  - pandas>=2.1
  - openpyxl>=3.0
  - scikit-learn>=1.3
+ - seaborn >= 0.12
 
  - pip:
  # code-checks
@@ -30,5 +31,6 @@ dependencies:
  - pandas-stubs>=2.1
  - ffmpeg-python>=0.2
  - tensorboard>=2.15
+ - tbparse >=0.0
  - torch-tb-profiler>=0.4
  - moviepy>=1.0
diff --git a/notebooks/.gitignore b/notebooks/.gitignore
@@ -0,0 +1,3 @@
+/predictions.pkl
+/tb_event_reader.pkl
+/result_data
diff --git a/notebooks/dataset_label_analysis.ipynb b/notebooks/dataset_label_analysis.ipynb
diff --git a/notebooks/predictions.pkl.dvc b/notebooks/predictions.pkl.dvc
@@ -0,0 +1,5 @@
+outs:
+- md5: 0a737bd16cabb291c83aac7510441e78
+ size: 6261785
+ hash: md5
+ path: predictions.pkl
diff --git a/notebooks/result_analysis.ipynb b/notebooks/result_analysis.ipynb
diff --git a/notebooks/result_data.dvc b/notebooks/result_data.dvc
@@ -0,0 +1,6 @@
+outs:
+- md5: fc1a513206db897593565f7024eba0e1.dir
+ size: 189619
+ nfiles: 8
+ hash: md5
+ path: result_data
diff --git a/notebooks/tb_event_reader.pkl.dvc b/notebooks/tb_event_reader.pkl.dvc
@@ -0,0 +1,5 @@
+outs:
+- md5: 948937e2cf861992f237e826454352d5
+ size: 8391171
+ hash: md5
+ path: tb_event_reader.pkl