diff --git a/ml_experiment/_utils/path.py b/ml_experiment/_utils/path.py new file mode 100644 index 0000000..eccc407 --- /dev/null +++ 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(), + )