Skip to content

Commit

Permalink
added exp
Browse files Browse the repository at this point in the history
  • Loading branch information
benhamner committed Jul 26, 2014
1 parent ce35116 commit 460c080
Showing 1 changed file with 17 additions and 0 deletions.
17 changes: 17 additions & 0 deletions experiments/regression_net_stop_iterations.jl
Original file line number Diff line number Diff line change
@@ -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)

0 comments on commit 460c080

Please sign in to comment.