From c8e7dbbf1d0ab660f4c3587d50111adf6a18b4e9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Pawe=C5=82=20Dawczak?= <pawel.dawczak@gmail.com> Date: Thu, 4 Apr 2019 21:55:50 +0100 Subject: [PATCH] hotfix/pd-fix-ruby-methods-argument: Fix argument name in generated method name (#49) --- .../DecisionTreeClassifier/templates/ruby/embedded.method.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sklearn_porter/estimator/classifier/DecisionTreeClassifier/templates/ruby/embedded.method.txt b/sklearn_porter/estimator/classifier/DecisionTreeClassifier/templates/ruby/embedded.method.txt index d751203b..d5781990 100644 --- a/sklearn_porter/estimator/classifier/DecisionTreeClassifier/templates/ruby/embedded.method.txt +++ b/sklearn_porter/estimator/classifier/DecisionTreeClassifier/templates/ruby/embedded.method.txt @@ -1,7 +1,7 @@ -def self.{method_name} (atts) +def self.{method_name} (features) classes = Array.new({n_classes}, 0) {branches} pos = classes.each_with_index.select {{|e, i| e==classes.max}}.map &:last return pos.min -end \ No newline at end of file +end