lib/naivebayes/classifier.rb in naivebayes-0.0.1 vs lib/naivebayes/classifier.rb in naivebayes-0.0.2
- old
+ new
@@ -1,10 +1,10 @@
#!/usr/bin/env ruby
# -*- coding: utf-8 -*-
module NaiveBayes
class Classifier
- def initialize(params)
+ def initialize(params = {})
@frequency_table = Hash.new
@word_table = Hash.new
@instance_count_of = Hash.new(0)
@total_count = 0
@model = params[:model]