baby_face 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: d3c03bdf7d6799c174a0c9c535c6fab41f18bccb
+  data.tar.gz: a6919788bac4954f5c982c2ca359e492c5bdfae9
+SHA512:
+  metadata.gz: 20a084a3ff38c57b8d480a396a11592285287b632c61172f444bb74a4247bf763865c15a916bdf466d3dcd08d093a415ed7460d00d55d7e7caf0e915fd5de036
+  data.tar.gz: 2c4955b4aff1056b0022a21f3838f771aa20f698b34007b7d78024873ca89ee5a483a0788e90d1f428e4ebd6aa3d57734e66fdf498afe8efe1d1ec50b8bde788

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in baby_face.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Yoshiori SHOJI
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,94 @@
+# BabyFace
+
+baby_face is a simple machine learning module.
+
+![](http://farm3.staticflickr.com/2835/11172696593_2e98d981d0_o.jpg)
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'baby_face'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install baby_face
+
+## Usage
+
+### target model( Poro, ActiveModel, ...)
+```ruby
+class Jedi
+  include BabyFace
+  acts_as_babyface features: :name,
+                   categories: [:light_side, :dark_side]
+
+  attr_accessor :name
+  def initialize(name)
+    @name = name
+  end
+end
+```
+
+### training
+
+```ruby
+Jedi.new("Anakin Skywalker").babyface.train_light_side
+Jedi.new("Darth Maul").babyface.train_dark_side
+```
+
+### classify
+
+```ruby
+Jedi.new("Luke Skywalker").babyface.light_side? # => true
+Jedi.new("Darth Vader").babyface.dark_side? # => true
+```
+
+### save training data
+
+```ruby
+BabyFace.configuration.data_dir = "/tmp/baby_face"
+Jedi.new("Luke Skywalker").babyface.save # => /tmp/baby_face/jedi.babyface
+```
+
+
+## more
+### tokenizer
+default : String#split
+
+```ruby
+acts_as_babyface features: :name,
+                 categories: [:ham, :spam],
+                 tokenizer: ->(text) {[text.upcase]}
+
+```
+
+### nested object
+support nested BabyFace object, array and hash.
+
+```ruby
+class Entry < ActiveRecord::Base
+  has_many :comments
+  include BabyFace
+  acts_as_babyface features: :title, :body, :comments,
+                   categories: [:ham, :spam]
+end
+
+class Comment < ActiveRecord::Base
+  include BabyFace
+  acts_as_babyface features: :title, :message
+end
+```
+
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/baby_face.gemspec

@@ -0,0 +1,26 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'baby_face/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "baby_face"
+  spec.version       = BabyFace::VERSION
+  spec.authors       = ["Yoshiori SHOJI"]
+  spec.email         = ["yoshiori@gmail.com"]
+  spec.description   = "baby_face is a simple machine learning module."
+  spec.summary       = spec.description
+  spec.homepage      = "https://github.com/yoshiori/baby_face"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_runtime_dependency "classifier"
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "rspec"
+end

data/bin/baby_face

@@ -0,0 +1,3 @@
+#!/usr/bin/env ruby
+
+require 'baby_face'

data/lib/baby_face/configuration.rb

@@ -0,0 +1,9 @@
+require "singleton"
+
+module BabyFace
+  class Configuration
+    include Singleton
+
+    attr_accessor :data_dir
+  end
+end

data/lib/baby_face/stand.rb

@@ -0,0 +1,92 @@
+require "baby_face/configuration"
+require 'pstore'
+require 'classifier'
+
+class BabyFace::Stand
+  def initialize(mod)
+    @mod = mod
+    @categories = mod.class.class_variable_get(:@@_categories)
+    @categories.each do |category|
+      self.class.class_eval do
+        define_method("#{category}?") {
+          bayes.classify(to_feature).gsub(" ", "_").downcase == category.to_s
+        }
+
+        define_method("train_#{category}") {
+          bayes.train(category, to_feature)
+        }
+      end
+    end
+    @tokenizer = mod.class.class_variable_get(:@@_tokenizer)
+  end
+
+  def to_feature
+    def scan(prefix, obj)
+      obj.class.class_variable_get(:@@_features).map do |attr|
+        _prefix = prefix.nil? ? attr : "#{prefix}_#{attr}"
+        value = obj.send(attr)
+        if value.is_a? BabyFace
+          scan(_prefix, value)
+        elsif value.is_a? Array
+          value.map do |val|
+            scan(_prefix, val)
+          end
+        elsif value.is_a? Hash
+          value.map do |key, val|
+            scan("#{_prefix}_#{key}", val)
+          end
+        else
+          wakachi(value.to_s).map do |text|
+            "#{_prefix}_#{text}"
+          end
+        end
+      end
+    end
+    scan(nil, @mod).flatten.join(" ")
+  end
+
+  def wakachi(text)
+    @tokenizer ? @tokenizer.call(text) : text.split
+  end
+
+  def save
+    bayes # Memoization (first access is in transaction)
+    pstore.transaction do
+      pstore[@mod.class.name] = bayes
+      pstore.commit
+    end
+  end
+
+  private
+  def data_path
+    if BabyFace.configuration.data_dir
+      Pathname.new(BabyFace.configuration.data_dir).join("#{@mod.class.name.downcase}.babyface")
+    else
+      Pathname.new("#{@mod.class.name.downcase}.babyface")
+    end
+  end
+
+  def pstore
+    if @mod.class.class_variable_defined?(:@@_pstore)
+      @mod.class.class_variable_get(:@@_pstore)
+    else
+      @mod.class.class_variable_set(:@@_pstore, PStore.new(data_path))
+    end
+  end
+
+  def bayes
+    if @mod.class.class_variable_defined?(:@@_bayes)
+      @mod.class.class_variable_get(:@@_bayes)
+    else
+      @mod.class.class_variable_set(:@@_bayes,
+        if data_path && data_path.exist?
+          pstore.transaction(true) do
+            pstore[@mod.class.name]
+          end
+        else
+          ::Classifier::Bayes.new *@categories
+        end
+      )
+    end
+  end
+end

data/lib/baby_face/version.rb

@@ -0,0 +1,3 @@
+module BabyFace
+  VERSION = "0.0.1"
+end

data/lib/baby_face.rb

@@ -0,0 +1,27 @@
+require "baby_face/stand"
+require "baby_face/configuration"
+require "baby_face/version"
+
+module BabyFace
+
+  def self.configuration
+    @configuration ||= Configuration.instance
+  end
+
+  def BabyFace.included(mod)
+    mod.define_singleton_method(
+      :acts_as_babyface,
+      lambda { |config|
+        features = *config[:features]
+        categories = *config[:categories]
+        mod.class_variable_set(:@@_features, features)
+        mod.class_variable_set(:@@_categories, categories)
+        mod.class_variable_set(:@@_tokenizer, config[:tokenizer])
+      }
+    )
+  end
+
+  def babyface
+    @_babyface ||= BabyFace::Stand.new(self)
+  end
+end

data/spec/baby_face/stand_spec.rb

@@ -0,0 +1,109 @@
+require "baby_face"
+
+describe BabyFace::Stand do
+
+  describe "#to_feature" do
+    subject { target.babyface.to_feature }
+
+    class Hoge
+      include BabyFace
+      attr_accessor :title, :name
+      acts_as_babyface features: [:title, :name]
+
+      def initialize(title: "Revenge of the Sith", name: "Anakin Skywalker")
+        @title = title
+        @name = name
+      end
+    end
+
+    class Bar
+      include BabyFace
+      attr_accessor :attr1, :attr2
+      acts_as_babyface features: [:attr1, :attr2]
+
+      def initialize(attr1: "foo", attr2: "bar")
+        @attr1 = attr1
+        @attr2 = attr2
+      end
+    end
+
+    context "Simple object" do
+      let(:target) { Hoge.new }
+
+      it { should eq "title_Revenge title_of title_the title_Sith name_Anakin name_Skywalker" }
+    end
+
+    context "Nested object" do
+      let(:target) { Hoge.new(name: Bar.new) }
+
+      it { should eq "title_Revenge title_of title_the title_Sith name_attr1_foo name_attr2_bar" }
+    end
+
+    context "Nested array" do
+      let(:target) { Hoge.new(name: [Bar.new, Bar.new]) }
+
+      it { should eq "title_Revenge title_of title_the title_Sith name_attr1_foo name_attr2_bar name_attr1_foo name_attr2_bar" }
+    end
+
+    context "Nested hash" do
+      let(:target) { Hoge.new(name: {bar1: Bar.new,bar2: Bar.new}) }
+
+      it { should eq "title_Revenge title_of title_the title_Sith name_bar1_attr1_foo name_bar1_attr2_bar name_bar2_attr1_foo name_bar2_attr2_bar" }
+    end
+  end
+
+  describe "#wakachi" do
+    context "Default" do
+      class Dummy
+        include BabyFace
+        attr_accessor :name
+        acts_as_babyface features: :name
+      end
+      let(:babyface) { Dummy.new.babyface }
+
+      it { expect(babyface.wakachi("aaa bbb")).to eq ["aaa", "bbb"] }
+    end
+
+    context "Use tokenizer" do
+      class Dummy2
+        include BabyFace
+        attr_accessor :name
+        acts_as_babyface features: :name,
+                         tokenizer: ->(text) {[text.upcase]}
+
+      end
+      let(:babyface) { Dummy2.new.babyface }
+
+      it { expect(babyface.wakachi("aaa bbb")).to eq ["AAA BBB"] }
+    end
+  end
+
+  describe "#train" do
+    class Jedi
+      include BabyFace
+      attr_accessor :name
+      acts_as_babyface features: :name,
+                       categories: [:light_side, :dark_side]
+
+      def initialize(name)
+        @name = name
+      end
+    end
+
+    it 'train jedi' do
+      10.times {
+        Jedi.new("Anakin Skywalker").babyface.train_light_side
+      }
+
+      10.times {
+        Jedi.new("Darth Maul").babyface.train_dark_side
+      }
+
+      luke = Jedi.new("Luke Skywalker")
+      vader = Jedi.new("Darth Vader")
+
+      expect(luke.babyface).to be_light_side
+      expect(vader.babyface).to be_dark_side
+    end
+  end
+end

metadata

@@ -0,0 +1,114 @@
+--- !ruby/object:Gem::Specification
+name: baby_face
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Yoshiori SHOJI
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-12-03 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: classifier
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  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'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  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: baby_face is a simple machine learning module.
+email:
+- yoshiori@gmail.com
+executables:
+- baby_face
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- baby_face.gemspec
+- bin/baby_face
+- lib/baby_face.rb
+- lib/baby_face/configuration.rb
+- lib/baby_face/stand.rb
+- lib/baby_face/version.rb
+- spec/baby_face/stand_spec.rb
+homepage: https://github.com/yoshiori/baby_face
+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.0.0
+signing_key: 
+specification_version: 4
+summary: baby_face is a simple machine learning module.
+test_files:
+- spec/baby_face/stand_spec.rb