nimbus 1.0 → 1.0.1
Sign up to get free protection for your applications and to get access to all the features.
- data/lib/nimbus/application.rb +2 -2
- data/lib/nimbus/forest.rb +1 -1
- metadata +1 -1
data/lib/nimbus/application.rb
CHANGED
@@ -93,7 +93,7 @@ module Nimbus
|
|
93
93
|
def output_tree_errors_file(forest)
|
94
94
|
File.open(@config.output_tree_errors_file , 'w') {|f|
|
95
95
|
1.upto(forest.tree_errors.size) do |te|
|
96
|
-
f.write("generalization error for tree
|
96
|
+
f.write("generalization error for tree #{te}: #{forest.tree_errors[te-1].round(5)}\n")
|
97
97
|
end
|
98
98
|
}
|
99
99
|
Nimbus.message "* Generalization errors for every tree saved to:"
|
@@ -126,7 +126,7 @@ module Nimbus
|
|
126
126
|
def output_snp_importances_file(forest)
|
127
127
|
File.open(@config.output_snp_importances_file , 'w') {|f|
|
128
128
|
forest.snp_importances.sort.each{|p|
|
129
|
-
f.write("SNP
|
129
|
+
f.write("SNP #{p[0]}: #{p[1].round(5)}\n")
|
130
130
|
}
|
131
131
|
}
|
132
132
|
Nimbus.message "* SNP importances for the forest saved to:"
|
data/lib/nimbus/forest.rb
CHANGED
@@ -41,7 +41,7 @@ module Nimbus
|
|
41
41
|
tree = Tree.new @options.tree
|
42
42
|
@trees << tree.seed(@options.training_set.individuals, tree_individuals_bag, @options.training_set.ids_fenotypes)
|
43
43
|
@tree_errors << tree.generalization_error_from_oob(tree_out_of_bag)
|
44
|
-
@tree_snp_importances << tree.estimate_importances(tree_out_of_bag)
|
44
|
+
@tree_snp_importances << tree.estimate_importances(tree_out_of_bag)
|
45
45
|
acumulate_predictions tree.predictions
|
46
46
|
Nimbus.clear_line!
|
47
47
|
end
|