[WIP] Add README file to project

.gitignore

@@ -1,10 +1,18 @@
+data/
+logs/
+models_checkpoints/
+predictions_plots/
+
+
+
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[codz]
 *$py.class
 
 # C extensions
 *.so
+!matlabfiles/codegen/lib/quadcopter_model/libquadcopter_model.so
 
 # Distribution / packaging
 .Python
@@ -14,7 +22,6 @@ dist/
 downloads/
 eggs/
 .eggs/
-lib/
 lib64/
 parts/
 sdist/
@@ -213,4 +220,12 @@ marimo/_lsp/
 __marimo__/
 
 # Streamlit
-.streamlit/secrets.toml
+.streamlit/secrets.toml
+
+
+**.ckpt
+models/accel_vs_voltage
+logs
+reports
+
+*.prof

.python-version

@@ -0,0 +1 @@
+3.11.13

.vscode/launch.json

@@ -0,0 +1,37 @@
+{
+    "version": "0.2.0",
+    "configurations": [
+        {
+            "name": "Python: Current File",
+            "type": "debugpy",
+            "request": "launch",
+            "program": "${file}",
+            "console": "integratedTerminal",
+            "justMyCode": true
+        },
+        {
+            "name": "Python: run_optim.py",
+            "type": "debugpy",
+            "request": "launch",
+            "program": "${workspaceFolder}/run_optim.py",
+            "console": "integratedTerminal",
+            "justMyCode": true
+        },
+        {
+            "name": "Python: run_model.py",
+            "type": "debugpy",
+            "request": "launch",
+            "program": "${workspaceFolder}/run_model.py",
+            "console": "integratedTerminal",
+            "justMyCode": true
+        },
+        {
+            "name": "Python: main.py",
+            "type": "debugpy",
+            "request": "launch",
+            "program": "${workspaceFolder}/main.py",
+            "console": "integratedTerminal",
+            "justMyCode": true
+        }
+    ]
+}

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+    "python-envs.pythonProjects": []
+}

config.yaml

@@ -1,9 +1,71 @@
 scenarios:
+  # - name: "control_4_motors"
+  #   data:
+  #     input_file: "data/control_4_motors/_inputs.csv"
+  #     states_file: "data/control_4_motors/_states.csv"
+  #     derivatives_file: "data/control_4_motors/_derivatives.csv"
+  #     output_file: "data/control_4_motors/out.csv"
+
+  #     #processed_file: "data/processed/accel_vs_voltage.npz"
+  #     input_dim: 4
+  #     states_dim: 12
+  #     derivatives_dim : 12
+  #     output_dim : 12
+  #     #use_simulink: false
+  #     #simulink_model: "UAV_model.slx"
+  #     #extra_simulink_samples: 0
+  #   preprocessing:
+  #     outlier_method: "iqr"  # options: iqr, zscore
+  #     normalize: true
+  #     normalization_scope: "per_feature"  # options: per_feature, global
+  #     fill_method: "linear"  # options: linear, cubic, nearest
+  #   augmentation:
+  #     noise_std: 0.01
+  #     window_size: 50
+  #     window_stride: 10
+  #   training:
+  #     models: ["gru"]
+  #     epochs: 1000
+  #     batch_size: 128
+  #     lr: 0.1
+  #     val_split: 0.2
+  #     test_split: 0.1
+  #     seed: 42
+  #     early_stopping_patience: 10
+  #     model_params:
+  #       gru:
+  #         hidden_dim: 16
+  #         num_layers: 2
+  #         dropout: 0.0
+  #       rnn:
+  #         hidden_dim: 16
+  #         num_layers: 1
+  #         dropout: 0.0
+  #     scheduler:
+  #       type: "ReduceLROnPlateau"
+  #       params:
+  #         patience: 10
+  #         factor: 0.5
+  #         min_lr: 0.000001
+  #   control:
+  #     methods: ["mpc"]
+  #     horizon: 10
+  #     rl_algorithm: "ppo"  # options: ppo, ddpg
+  #     mpc_weight_tracking: 1.0
+  #     mpc_weight_smoothness: 0.1
+  #     mpc_weight_effort: 0.01
+  #   reporting:
+  #     generate_pdf: true
+  #     include_plots: true
+  #     metrics_file: "results/control_4_motors/metrics.json"
+
   - name: "accel_vs_voltage"
     data:
       input_file: "data/accel_vs_voltage/in.csv"
       output_file: "data/accel_vs_voltage/out.csv"
       processed_file: "data/processed/accel_vs_voltage.npz"
+      input_dim: 1
+      output_dim: 1
       use_simulink: false
       simulink_model: "UAV_model.slx"
       extra_simulink_samples: 0
@@ -16,15 +78,34 @@ scenarios:
       noise_std: 0.01
       window_size: 50
       window_stride: 10
-      simulink_noise_samples: 0
     training:
-      models: ["linear", "gru"]
-      epochs: 50
-      batch_size: 32
-      lr: 0.001
+      models: ["gru"]
+      epochs: 1000
+      batch_size: 128
+      lr: 0.01
       val_split: 0.2
       test_split: 0.1
       seed: 42
+      early_stopping_patience: 10
+      model_params:
+        gru:
+          hidden_dim: 32
+          num_layers: 2
+          dropout: 0.0
+        lstm:
+          hidden_dim: 16
+          num_layers: 1
+          dropout: 0.0
+        rnn:
+          hidden_dim: 8
+          num_layers: 1
+          dropout: 0.0
+      scheduler:
+        type: "ReduceLROnPlateau"
+        params:
+          patience: 10
+          factor: 0.5
+          min_lr: 0.000001
     control:
       methods: ["mpc"]
       horizon: 10

logs/accel_vs_voltage/linear/version_0/hparams.yaml

@@ -0,0 +1,3 @@
+input_dim: 2000
+lr: 0.001
+output_dim: 2000

logs/accel_vs_voltage/version_0/hparams.yaml

@@ -0,0 +1,6 @@
+dropout: 0.2
+hidden_dim: 64
+input_dim: 2000
+lr: 0.001
+num_layers: 2
+output_dim: 2000