Skip to content
Snippets Groups Projects
run_experiments.jl 852 B
Newer Older
include("setup_env.jl");
include("experiment.jl");
Pat Alt's avatar
Pat Alt committed

# User inputs:
pat-alt's avatar
pat-alt committed
if "run-all" in ARGS
    datanames = ["linearly_separable", "moons", "circles", "mnist", "gmsc"]
else
pat-alt's avatar
pat-alt committed
    datanames = [ARGS[findall(contains.(ARGS, "data="))] |> x -> replace(x, "data=" => "")]
Pat Alt's avatar
Pat Alt committed

# Linearly Separable
if "linearly_separable" in datanames
Pat Alt's avatar
Pat Alt committed
    @info "Running linearly separable experiment."
    include("linearly_separable.jl")
end

# Moons
if "moons" in datanames
Pat Alt's avatar
Pat Alt committed
    @info "Running moons experiment."
    include("moons.jl")
end

# Circles
if "circles" in datanames
Pat Alt's avatar
Pat Alt committed
    @info "Running circles experiment."
    include("circles.jl")
end

# MNIST
if "mnist" in datanames
Pat Alt's avatar
Pat Alt committed
    @info "Running MNIST experiment."
    include("mnist.jl")
end

# GMSC
if "gmsc" in datanames
Pat Alt's avatar
Pat Alt committed
    @info "Running GMSC experiment."
    include("gmsc.jl")
end
Pat Alt's avatar
Pat Alt committed

if PARALLEL
    MPI.Finalize()
end