Add Ruby variation of the decision tree classifier

readme.md

@@ -61,7 +61,7 @@ Transpile trained [scikit-learn](https:/scikit-learn/scikit-learn) e
   <td align="center"><a href="examples/classifier/DecisionTreeClassifier/js/basics.ipynb">✓</a></td>
   <td align="center"></td>
   <td align="center"><a href="examples/classifier/DecisionTreeClassifier/php/basics.ipynb">✓</a></td>
-  <td align="center"></td>
+  <td align="center">✓</td>
   </tr>
   <tr>
   <td><a href="http://scikit-learn.org/0.18/modules/generated/sklearn.ensemble.RandomForestClassifier.html">sklearn.ensemble.RandomForestClassifier</a></td>

sklearn_porter/classifier/DecisionTreeClassifier/__init__.py

@@ -47,6 +47,18 @@ class DecisionTreeClassifier(Classifier):
  'arr': '$classes[{0}] = {1}',
  'indent': ' ',
  'join': '; ',
+ },
+ 'ruby': {
+ # 'init': '{name} = {value}',
+ # 'type': '{0}',
+ 'if': 'if atts[{0}] {1} {2}',
+ 'else': 'else',
+ 'endif': 'end',
+ 'arr': 'classes[{0}] = {1}',
+ # 'arr[]': '{name} = [{values}]',
+ # 'arr[][]': '{name} = [{values}]',
+ 'indent': ' ',
+ 'join': ' ',
  }
  }
  # @formatter:on
@@ -179,7 +191,7 @@ def create_tree(self):
  if self.n_features > 1 or (self.n_features == 1 and i >= 0):
  feature_indices.append([str(j) for j in range(n_features)][i])
 
- indentation = 1 if self.target_language in ['java', 'js', 'php'] else 0
+ indentation = 1 if self.target_language in ['java', 'js', 'php', 'ruby'] else 0
  return self.create_branches(
  self.model.tree_.children_left,
  self.model.tree_.children_right,
@@ -196,7 +208,7 @@ def create_method(self, class_name, method_name):
  :return out : string
  The built method as string.
  """
- n_indents = 1 if self.target_language in ['java', 'js', 'php'] else 0
+ n_indents = 1 if self.target_language in ['java', 'js', 'php', 'ruby'] else 0
  branches = self.indent(self.create_tree(), n_indents=1)
  temp_method = self.temp('method', n_indents=n_indents, skipping=True)
  out = temp_method.format(class_name=class_name, method_name=method_name,

sklearn_porter/classifier/DecisionTreeClassifier/templates/ruby/class.txt

@@ -0,0 +1,8 @@
+class {class_name}
+ {method}
+end
+
+if ARGV.length == {n_features}
+ atts = ARGV.collect {{ |i| i.to_f }}
+ puts {class_name}.{method_name}(atts)
+end

sklearn_porter/classifier/DecisionTreeClassifier/templates/ruby/method.txt

@@ -0,0 +1,7 @@
+def self.{method_name} (atts)
+ classes = Array.new({n_classes}, 0)
+ {branches}
+
+ pos = classes.each_with_index.select {{|e, i| e==classes.max}}.map &:last
+ return pos.min
+end

tests/classifier/DecisionTreeClassifier/DecisionTreeClassifierRubyTest.py

@@ -0,0 +1,18 @@
+# -*- coding: utf-8 -*-
+
+from unittest import TestCase
+
+from sklearn.tree import DecisionTreeClassifier
+
+from ..Classifier import Classifier
+from ...language.Ruby import Ruby
+
+
+class DecisionTreeClassifierRubyTest(Ruby, Classifier, TestCase):
+
+ def setUp(self):
+ super(DecisionTreeClassifierRubyTest, self).setUp()
+ self.mdl = DecisionTreeClassifier(random_state=0)
+
+ def tearDown(self):
+ super(DecisionTreeClassifierRubyTest, self).tearDown()