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