feat: added utility function for getting results path

panahiparham · Oct 23, 2024 · ed08c84 · ed08c84
1 parent 002417f
commit ed08c84
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/ml_experiment/_utils/path.py b/ml_experiment/_utils/path.py
@@ -0,0 +1,12 @@
+import os
+
+def get_experiment_name():
+    import __main__
+    return __main__.__file__.split('/')[-2]
+
+def get_results_path(base_path: str) -> str:
+    return os.path.join(
+        base_path,
+        'results',
+        get_experiment_name(),
+    )