diff --git a/mu_map/random_search/eval/label_outliers.py b/mu_map/random_search/eval/label_outliers.py index b7dd78bed0e8954c09cda315c7388b7161e1485c..555aeab8167f48f5a9e0afaaddcbfb71d6b7e28c 100644 --- a/mu_map/random_search/eval/label_outliers.py +++ b/mu_map/random_search/eval/label_outliers.py @@ -9,7 +9,7 @@ from mu_map.dataset.default import MuMapDataset from mu_map.dataset.transform import SequenceTransform, PadCropTranform from mu_map.models.unet import UNet from mu_map.random_search.cgan import load_params -from mu_map.random_search.show_predictions import main +from mu_map.random_search.eval.show_predictions import main from mu_map.random_search.eval.util import load_data controls = """ @@ -71,7 +71,6 @@ for i, run in enumerate(runs): data["run"].append(int(run)) data["outlier"].append(False) - dir_run = os.path.join(args.random_search_dir, runs[run]["dir"]) params = runs[run]["params"]