tree_thinking 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 12adb5716b0ac77981cfcd8cae52860a948e01f9
+  data.tar.gz: 71a92dca72fb0a22e5b6810adb2818d6cb9b0b6b
+SHA512:
+  metadata.gz: 3468cb179c74602796a835f7a86ba0465f0c11ad08d353d466504917f17a34877672bf02e7777d5c4ccea2c4887987161300acd8a6c74ee6ae25aef72e47e399
+  data.tar.gz: 4e6cbdaddfb3ffe19de423c16abbc6ed072ca83abd3514fedb44e6d069d5aba69dfaebaae35878fba1010dd61cdbf8e12e6e50866a1f7153a0670496ae329f89

data/.gitignore

@@ -0,0 +1,2 @@
+*.gem
+.DS_Store

data/Gemfile

@@ -0,0 +1,3 @@
+source 'https://rubygems.org'
+
+gemspec

data/Gemfile.lock

@@ -0,0 +1,43 @@
+PATH
+  remote: .
+  specs:
+    tree_thinking (0.1.0)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    coderay (1.1.1)
+    diff-lcs (1.2.5)
+    method_source (0.8.2)
+    pry (0.10.4)
+      coderay (~> 1.1.0)
+      method_source (~> 0.8.1)
+      slop (~> 3.4)
+    rake (10.5.0)
+    rspec (3.5.0)
+      rspec-core (~> 3.5.0)
+      rspec-expectations (~> 3.5.0)
+      rspec-mocks (~> 3.5.0)
+    rspec-core (3.5.4)
+      rspec-support (~> 3.5.0)
+    rspec-expectations (3.5.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.5.0)
+    rspec-mocks (3.5.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.5.0)
+    rspec-support (3.5.0)
+    slop (3.6.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  bundler (~> 1.13)
+  pry
+  rake (~> 10.0)
+  rspec (~> 3.0)
+  tree_thinking!
+
+BUNDLED WITH
+   1.13.1

data/README.md

@@ -0,0 +1,27 @@
+# The Ruby Gem
+
+The ruby gem currently supports loading and applying a YAML decision tree.  The
+following example is from the tests and shows how a YAML fixture file is parsed
+into a `BinaryTree` object:
+
+```ruby
+tree = TreeThinking::BinaryTreeFactory.from_yaml('./spec/fixtures/simple_tree.yaml')
+```
+
+This tree can then be used to get a probability from an answer vector:
+
+```ruby
+tree.call([1, 0])
+#> 0.1
+```
+
+In this case, the answer vector is `[1,0]`, however another approach that can
+be used is a `Struct` where the attribute order matches the order of the
+features in the decision tree.  For example:
+
+```ruby
+Answer = Struct.new(:likes_ice_cream, :likes_chocolate)
+answer_vector = Answer.new(1,0)
+tree.call(answer_vector)
+#> 0.1
+```

data/lib/tree_thinking.rb

@@ -0,0 +1,6 @@
+require "tree_thinking/version"
+require "tree_thinking/binary_tree"
+require "tree_thinking/binary_tree_factory"
+
+module TreeThinking
+end

data/lib/tree_thinking/binary_tree.rb

@@ -0,0 +1,36 @@
+module TreeThinking
+  class BinaryTree
+    attr_reader :name
+    attr_accessor :tree
+
+    Node = Struct.new(:feature_idx, :operator, :threshold, :probablity, :results)
+
+    def initialize(name: nil)
+      @name = name
+    end
+
+    def call(answer_vector)
+      node = self.tree
+
+      while node.probablity.nil? do
+        node = node.results.fetch(decision(node, answer_vector))
+      end
+
+      node.probablity
+    end
+
+    private
+
+    def decision(node, answer_vector)
+      case node.operator
+      when '=' then answer_vector[node.feature_idx] == node.threshold
+      when '>' then answer_vector[node.feature_idx] > node.threshold
+      when '<' then answer_vector[node.feature_idx] < node.threshold
+      when '>=' then answer_vector[node.feature_idx] >= node.threshold
+      when '<=' then answer_vector[node.feature_idx] <= node.threshold
+      else
+        answer_vector[node.feature_idx] <= node.threshold
+      end
+    end
+  end
+end

data/lib/tree_thinking/binary_tree_factory.rb

@@ -0,0 +1,35 @@
+require 'yaml'
+
+module TreeThinking
+  class BinaryTreeFactory
+    attr_reader :tree
+
+    def self.from_yaml(path)
+      new(YAML.load(File.read(path))).build!
+    end
+
+    def initialize(tree)
+      @tree = tree
+    end
+
+    def build!
+      btree = BinaryTree.new(name: tree['class_name'])
+      btree.tree = nodify(tree['tree'])
+      btree
+    end
+
+    private
+
+    def nodify(attrs)
+      BinaryTree::Node.new(
+        attrs['feature_idx'],
+        attrs['op'],
+        attrs['thr'],
+        attrs['prob'],
+        Hash[attrs.fetch('results', {}).map do |return_val, child_node|
+          [return_val, nodify(child_node)]
+        end]
+      )
+    end
+  end
+end

data/lib/tree_thinking/version.rb

@@ -0,0 +1,3 @@
+module TreeThinking
+  VERSION = "0.1.0"
+end

data/tree_thinking.gemspec

@@ -0,0 +1,28 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'tree_thinking/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "tree_thinking"
+  spec.version       = TreeThinking::VERSION
+  spec.authors       = ["Greggory Rothmeier"]
+  spec.email         = ["greggroth@gmail.com"]
+
+  spec.summary       = %q{Language-agnostic binary tree representation parsing and executation}
+  spec.description   = %q{Language-agnostic binary tree representation parsing and executation}
+  spec.homepage      = ""
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject do |f|
+    f.match(%r{^(test|spec|features)/})
+  end
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.13"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rspec", "~> 3.0"
+  spec.add_development_dependency "pry"
+end

metadata

@@ -0,0 +1,110 @@
+--- !ruby/object:Gem::Specification
+name: tree_thinking
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Greggory Rothmeier
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2016-12-28 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.13'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.13'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Language-agnostic binary tree representation parsing and executation
+email:
+- greggroth@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- Gemfile.lock
+- README.md
+- lib/tree_thinking.rb
+- lib/tree_thinking/binary_tree.rb
+- lib/tree_thinking/binary_tree_factory.rb
+- lib/tree_thinking/version.rb
+- tree_thinking.gemspec
+homepage: ''
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: Language-agnostic binary tree representation parsing and executation
+test_files: []
+has_rdoc: