Skip to content
Snippets Groups Projects
Commit fa0bf925 authored by pat-alt's avatar pat-alt
Browse files

now looking good?

parent b75dc906
No related branches found
No related tags found
1 merge request!7669 initial run including fmnist lenet and new method
This diff is collapsed.
...@@ -5,5 +5,5 @@ counterfactual_data, test_data = train_test_split( ...@@ -5,5 +5,5 @@ counterfactual_data, test_data = train_test_split(
) )
run_experiment( run_experiment(
counterfactual_data, test_data; counterfactual_data, test_data;
dataname="Linearly Separable" dataname="Linearly Separable",
) )
\ No newline at end of file
...@@ -2,10 +2,13 @@ include("setup_env.jl"); ...@@ -2,10 +2,13 @@ include("setup_env.jl");
# User inputs: # User inputs:
if "run-all" in ARGS if "run-all" in ARGS
datanames = ["linearly_separable", "moons", "circles", "mnist", "gmsc"] datanames = ["linearly_separable", "moons", "circles", "mnist", "fmnist", "gmsc"]
else elseif any(contains.(ARGS, "data="))
datanames = [ARGS[findall(contains.(ARGS, "data="))][1] |> x -> replace(x, "data=" => "")] datanames = [ARGS[findall(contains.(ARGS, "data="))][1] |> x -> replace(x, "data=" => "")]
datanames = replace.(split(datanames[1], ","), " " => "") datanames = replace.(split(datanames[1], ","), " " => "")
else
@warn "No dataset specified, defaulting to all."
datanames = ["linearly_separable", "moons", "circles", "mnist", "fmnist", "gmsc"]
end end
# Linearly Separable # Linearly Separable
......
...@@ -72,6 +72,9 @@ time_stamped = false ...@@ -72,6 +72,9 @@ time_stamped = false
if any(contains.(ARGS, "output_path")) if any(contains.(ARGS, "output_path"))
@assert sum(contains.(ARGS, "output_path")) == 1 "Only one output path can be specified." @assert sum(contains.(ARGS, "output_path")) == 1 "Only one output path can be specified."
_path = ARGS[findall(contains.(ARGS, "output_path"))][1] |> x -> replace(x, "output_path=" => "") _path = ARGS[findall(contains.(ARGS, "output_path"))][1] |> x -> replace(x, "output_path=" => "")
elseif isinteractive()
@info "You are running experiments interactively. By default, results will be saved in a temporary directory."
_path = tempdir()
else else
timestamp = Dates.format(now(), "yyyy-mm-dd@HH:MM") timestamp = Dates.format(now(), "yyyy-mm-dd@HH:MM")
time_stamped = true time_stamped = true
...@@ -82,8 +85,6 @@ end ...@@ -82,8 +85,6 @@ end
const DEFAULT_OUTPUT_PATH = _path const DEFAULT_OUTPUT_PATH = _path
const TIME_STAMPED = time_stamped const TIME_STAMPED = time_stamped
ispath(DEFAULT_OUTPUT_PATH) || mkpath(DEFAULT_OUTPUT_PATH)
"Boolean flag to only train models." "Boolean flag to only train models."
const ONLY_MODELS = "only_models" ARGS ? true : false const ONLY_MODELS = "only_models" ARGS ? true : false
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment