lib/lightgbm.rb in lightgbm-0.1.9 vs lib/lightgbm.rb in lightgbm-0.2.0
- old
+ new
@@ -59,20 +59,18 @@
if valid_sets.any?
# print results
messages = []
+ eval_valid = booster.eval_valid
if valid_contain_train
- # not sure why reversed in output
- booster.eval_train.reverse.each do |res|
- messages << "%s's %s: %g" % [res[0], res[1], res[2]]
- end
+ eval_valid = eval_valid + booster.eval_train
end
-
- eval_valid = booster.eval_valid
# not sure why reversed in output
- eval_valid.reverse.each do |res|
+ eval_valid.reverse!
+
+ eval_valid.each do |res|
messages << "%s's %s: %g" % [res[0], res[1], res[2]]
end
message = "[#{iteration + 1}]\t#{messages.join("\t")}"
@@ -186,11 +184,10 @@
if early_stopping_rounds
# use best iteration from first metric if not stopped early
best_iteration ||= best_iter[best_iter.keys.first]
eval_hist.each_key do |k|
- # TODO uncomment for 0.2.0
- # eval_hist[k] = eval_hist[k].first(best_iteration + 1)
+ eval_hist[k] = eval_hist[k].first(best_iteration + 1)
end
end
eval_hist
end