From 1ee8cbc75c6f140afc57036f4812d3493cd81038 Mon Sep 17 00:00:00 2001
From: Niels Geens <niels.geens@materialise.be>
Date: Fri, 24 Jan 2025 10:41:56 +0100
Subject: [PATCH] Initial training config file

---
 training/config.yaml | 10 ++++++++++
 1 file changed, 10 insertions(+)
 create mode 100644 training/config.yaml

diff --git a/training/config.yaml b/training/config.yaml
new file mode 100644
index 0000000..ded2cab
--- /dev/null
+++ b/training/config.yaml
@@ -0,0 +1,10 @@
+model:
+  system_prompt_path: "system_prompt.md"
+  action_schema: "action_schema.xsd"
+params:
+  learning_rate: 1e-5
+  epochs: 3
+data:
+  - "training/clean_start/"
+  - "training/delete_indicated_entries/"
+  - "training/list_entries_to_delete/"
\ No newline at end of file