Sha256: b5669e389c9024cf30491d83e9c83223225bec5ee5381bebf26236c84d19494a
Contents?: true
Size: 951 Bytes
Versions: 4
Compression:
Stored size: 951 Bytes
Contents
require 'spec_helper'
describe ::Bae::NativeClassifier do
subject { described_class.new }
it "can classify a hash document" do
subject.train("positive", {"aaa" => 0, "bbb" => 1})
subject.train("negative", {"ccc" => 2, "ddd" => 3})
subject.finish_training!
results = subject.classify({"aaa" => 1, "bbb" => 1})
expect(results["positive"]).to be_within(0.001).of(0.94117)
expect(results["negative"]).to be_within(0.001).of(0.05882)
end
it "can classify from a string based document" do
subject.train("positive", "aaa aaa bbb");
subject.train("negative", "ccc ccc ddd ddd");
subject.train("neutral", "eee eee eee fff fff fff");
subject.finish_training!
results = subject.classify("aaa bbb")
expect(results["positive"]).to be_within(0.001).of(0.89626)
expect(results["negative"]).to be_within(0.001).of(0.06639)
expect(results["neutral"]).to be_within(0.001).of(0.03734)
end
end
Version data entries
4 entries across 4 versions & 1 rubygems