zydronium/neocities
1
0
Fork 0
mirror of https://github.com/neocities/neocities.git synced 2025-08-01 23:31:52 +02:00
Code Issues Projects Releases Packages Wiki Activity

experimental statistical language classification system

Browse source
This commit is contained in:
Kyle Drake 2015-12-28 14:32:43 -06:00
parent 9cc85a48bb
commit 242cc6e92d
15 changed files with 178 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

20
tests/site_file_tests.rb
View file

@ -329,5 +329,25 @@ describe 'site_files' do
upload 'files[]' => Rack::Test::UploadedFile.new('./tests/files/index.html', 'text/html')
@site.reload.changed_count.must_equal 2
end
describe 'classification' do
before do
$trainer.instance_variable_get('@db').redis.flushall
end
it 'trains files' do
upload 'files[]' => Rack::Test::UploadedFile.new('./tests/files/classifier/ham.html', 'text/html')
upload 'files[]' => Rack::Test::UploadedFile.new('./tests/files/classifier/spam.html', 'text/html')
upload 'files[]' => Rack::Test::UploadedFile.new('./tests/files/classifier/phishing.html', 'text/html')
@site.train 'ham.html'
@site.train 'spam.html', 'spam'
@site.train 'phishing.html', 'phishing'
@site.classify('ham.html').must_equal 'ham'
@site.classify('spam.html').must_equal 'spam'
@site.classify('phishing.html').must_equal 'phishing'
end
end
end
end