Roche · jpaillard · Dec 17, 2024 · Dec 12, 2024 · Dec 12, 2024
diff --git a/README.md b/README.md
@@ -1,3 +1,14 @@
+
+# GREEN architecture
+![alt text](assets/concept_figure.png)
+
+![CI](https://github.com/Roche/neuro-green/blob/main/.github/workflows/lint_and_test.yaml/badge.svg)
+
+## Getting started 
+
+```
+pip install -e .
+```
 ## Citation
 When using our code, please cite the reference article:
 

diff --git a/assets/concept_figure.png b/assets/concept_figure.png
diff --git a/research_code/__init__.py → green/research_code/__init__.py b/research_code/__init__.py → green/research_code/__init__.py
diff --git a/research_code/crossval_utils.py → green/research_code/crossval_utils.py b/research_code/crossval_utils.py → green/research_code/crossval_utils.py
diff --git a/research_code/example.ipynb → green/research_code/example.ipynb b/research_code/example.ipynb → green/research_code/example.ipynb
diff --git a/research_code/example_wo_wav.ipynb → green/research_code/example_wo_wav.ipynb b/research_code/example_wo_wav.ipynb → green/research_code/example_wo_wav.ipynb
diff --git a/research_code/pl_utils.py → green/research_code/pl_utils.py b/research_code/pl_utils.py → green/research_code/pl_utils.py
diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,7 @@
 torch
 geotorch
 mne
-
+lightning
+scikit-learn
+pandas
+pytest
diff --git a/research_code/requirements.txt b/research_code/requirements.txt