From 460c080b7e2fc77bc1714dbaeaeb4524ab73325e Mon Sep 17 00:00:00 2001 From: Ben Hamner Date: Fri, 25 Jul 2014 18:34:47 -0700 Subject: [PATCH] added exp --- experiments/regression_net_stop_iterations.jl | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 experiments/regression_net_stop_iterations.jl diff --git a/experiments/regression_net_stop_iterations.jl b/experiments/regression_net_stop_iterations.jl new file mode 100644 index 0000000..5e5602f --- /dev/null +++ b/experiments/regression_net_stop_iterations.jl @@ -0,0 +1,17 @@ +using Gadfly +using MachineLearning +using RDatasets + +opts_generator(stop_iterations::Int) = regression_net_options(hidden_layers=[500], stop_criteria=StopAfterIteration(stop_iterations), learning_rate=0.01) +opts_sweep = [20000:20000:100000] + +data = dataset("car", "Prestige") +data_generator(seed) = split_train_test(data, :Prestige, seed=seed) + +res = compare(data_generator, opts_generator, opts_sweep, cor) +draw(PNG("regression_net_stop_iterations.png", 8inch, 6inch), plot(res, x=:Name, y=:Score)) + +dist = by(res, :Name, df -> DataFrame(Mean=mean(df[:Score]), + Q1=quantile(df[:Score], 0.25), + Q3=quantile(df[:Score], 0.75))) +println(dist)