v0.1.6

ramiz khaled · ramiz khaled · commit f77f2773157a · 2026-04-14T09:44:20.000+02:00
diff --git a/amn/_version.py b/amn/_version.py
@@ -1 +1 @@
-__version__ = "0.1.5"
+__version__ = "0.1.6"
diff --git a/amn/predict.py b/amn/predict.py
@@ -60,6 +60,11 @@ def predict_amn(
         "biolog": {
             "cobra_file": get_default_model('biolog','build'),
             "metric": "r2_score"
+        },
+
+        "custom": {
+            "cobra_file": cobra_file,
+            "metric": metric
         }
     }
 
diff --git a/amn/train.py b/amn/train.py
@@ -59,6 +59,10 @@ def build_training_set(
 
         "biolog": {
             "cobra_file": get_default_model('biolog','build')
+        },
+
+        "custom": {
+            "cobra_file": cobra_file
         }
     }
     #if organism not in PRESETS:
@@ -147,6 +151,17 @@ def train_gr_prediction(
             "xfold": 5,
             "niter": 0,
             "batch_size": 100
+        },
+        
+        "custom": {
+            "trainingfile": trainingfile,
+            "cobraname_override": cobraname_override,
+            "n_hidden": n_hidden,
+            "hidden_dim": hidden_dim,
+            "epochs": epochs,
+            "xfold": xfold,
+            "niter": niter,
+            "batch_size": batch_size
         }
     }
     #if organism not in PRESETS:

Original file line number	Diff line number	Diff line change
`@@ -60,6 +60,11 @@ def predict_amn(`
`60`	`60`	`"biolog": {`
`61`	`61`	`"cobra_file": get_default_model('biolog','build'),`
`62`	`62`	`"metric": "r2_score"`
	`63`	`+ },`
	`64`	`+`
	`65`	`+ "custom": {`
	`66`	`+ "cobra_file": cobra_file,`
	`67`	`+ "metric": metric`
`63`	`68`	`}`
`64`	`69`	`}`
`65`	`70`