bm25f 0.1.0 → 0.2.0
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/test/test_bm25f.rb +23 -0
- metadata +16 -1
checksums.yaml
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
---
|
2
2
|
SHA256:
|
3
|
-
metadata.gz:
|
4
|
-
data.tar.gz:
|
3
|
+
metadata.gz: 2649ac16a2f5adeb5815b2f472e62f98aec6b05af882648e4d128051c5fb90e0
|
4
|
+
data.tar.gz: e70ee90881438943874b4ae5e4852a6e7d44eda53f0315af35ea85565a2cde70
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: 1f7a68810a77daa256eb3e3177748e6b22f53cf3dfd2e68c68e82c4b7c143cdc60ac38c0e519e6f4b3c90725123c62f2b5e852898230c5e9812ca181e3efa577
|
7
|
+
data.tar.gz: 6690099e9435eebc6676d7bbd0d5530218e2d360a2ca8fb3e5156faf74b426364765eab60bdab5bb2caa85deadb69bb22b951a987d9904eaf1cb75bbb1b31520
|
data/test/test_bm25f.rb
ADDED
@@ -0,0 +1,23 @@
|
|
1
|
+
require 'minitest/autorun'
|
2
|
+
require 'bm25f'
|
3
|
+
|
4
|
+
class BM25FTest < Minitest::Test
|
5
|
+
def setup
|
6
|
+
@bm25f = BM25F.new
|
7
|
+
@documents = [
|
8
|
+
{ title: 'hello world', content: 'foo bar baz' },
|
9
|
+
{ title: 'foo bar', content: 'goodbye, world!' }
|
10
|
+
]
|
11
|
+
end
|
12
|
+
|
13
|
+
def test_score
|
14
|
+
@bm25f.fit @documents
|
15
|
+
scores = @bm25f.score 'hello world foo bar baz'
|
16
|
+
|
17
|
+
# Sort
|
18
|
+
scores = scores.to_a.sort_by { |_, v| v.to_i }
|
19
|
+
|
20
|
+
# Checks if the most matching element is the first element
|
21
|
+
assert scores.last[0].zero?
|
22
|
+
end
|
23
|
+
end
|
metadata
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
--- !ruby/object:Gem::Specification
|
2
2
|
name: bm25f
|
3
3
|
version: !ruby/object:Gem::Version
|
4
|
-
version: 0.
|
4
|
+
version: 0.2.0
|
5
5
|
platform: ruby
|
6
6
|
authors:
|
7
7
|
- catflip
|
@@ -24,6 +24,20 @@ dependencies:
|
|
24
24
|
- - "~>"
|
25
25
|
- !ruby/object:Gem::Version
|
26
26
|
version: '2.1'
|
27
|
+
- !ruby/object:Gem::Dependency
|
28
|
+
name: rake
|
29
|
+
requirement: !ruby/object:Gem::Requirement
|
30
|
+
requirements:
|
31
|
+
- - ">="
|
32
|
+
- !ruby/object:Gem::Version
|
33
|
+
version: '0'
|
34
|
+
type: :development
|
35
|
+
prerelease: false
|
36
|
+
version_requirements: !ruby/object:Gem::Requirement
|
37
|
+
requirements:
|
38
|
+
- - ">="
|
39
|
+
- !ruby/object:Gem::Version
|
40
|
+
version: '0'
|
27
41
|
description: A fast implementation of the BM25F ranking algorithm for information
|
28
42
|
retrieval systems, written in Ruby.
|
29
43
|
email:
|
@@ -32,6 +46,7 @@ extensions: []
|
|
32
46
|
extra_rdoc_files: []
|
33
47
|
files:
|
34
48
|
- lib/bm25f.rb
|
49
|
+
- test/test_bm25f.rb
|
35
50
|
homepage: https://github.com/catflip/bm25f-ruby
|
36
51
|
licenses:
|
37
52
|
- AGPL-3.0
|