bayesball 0.1.0 → 0.1.1

data/README.md

@@ -1,6 +1,7 @@
 # Bayesball
 
-TODO: Write a gem description
+Bayesball is a naive Bayes classifier with support for in memory and mongo
+persistence.
 
 ## Installation
 
@@ -18,7 +19,13 @@ Or install it yourself as:
 
 ## Usage
 
-TODO: Write usage instructions here
+```ruby
+classifier = Bayesball::Classifier.new
+
+classifier.train 'basketball', 'dunk the ball'
+
+classifier.classify 'Slam Dunk!' #=> "basketball"
+```
 
 ## Contributing
 

data/Rakefile

@@ -2,6 +2,7 @@ require "bundler/gem_tasks"
 require 'rake/testtask'
 
 Rake::TestTask.new do |t|
+    puts ENV['TERM']
     t.libs = ['spec', 'lib']
-    t.test_files = Dir['spec/**/*_spec.rb']
+    t.pattern = 'spec/**/*_spec.rb'
 end

data/lib/bayesball/classifier.rb

@@ -36,7 +36,7 @@ module Bayesball
 
     def classify(payload)
       s = score(payload)
-      s.sort_by{|k,v| v}.reverse[0][0]
+      s.sort_by{|_,v| v}.map{|k,_| k}.reverse.first
     end
   end
 end

data/lib/bayesball/version.rb

@@ -1,3 +1,3 @@
 module Bayesball
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/spec/bayesball/classifier_spec.rb

@@ -1,7 +1,7 @@
 require 'spec_helper'
 
 describe Bayesball::Classifier do
-  let(:subject) { Bayesball::Classifier.new(Bayesball::Persistence::Mongo.new(MONGO_URI)) }
+  let(:subject) { Bayesball::Classifier.new }
 
   it 'should do word counts' do
     result = subject.word_counts('Hello, friend. Hello!')
@@ -47,4 +47,8 @@ describe Bayesball::Classifier do
     subject.classify('Hitting a ball made of rubber').must_equal 'racquetball'
     subject.classify('The winning team is kicking butt. They always make the ball go in the hoop every time').must_equal 'basketball'
   end
+
+  it 'should return nil if cannot classify' do
+    subject.classify('there are no topics').must_be_nil
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: bayesball
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-04-09 00:00:00.000000000 Z
+date: 2012-04-13 00:00:00.000000000 Z
 dependencies: []
 description: A bayes classifier
 email:
@@ -48,7 +48,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -3818961451437177514
+      hash: -4130593485830818059
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -57,7 +57,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -3818961451437177514
+      hash: -4130593485830818059
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.18