music_detector 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9d4347bc2aa7368840aeacbfea0d5f8988198d96
-  data.tar.gz: b91ba65982e25f962fe90da3714ce9a55f145c0b
+  metadata.gz: 3401a5f37951bc70e14a7f2c09f8b1166cefd8c7
+  data.tar.gz: f129f4889bd70baed3d2a2982e4055e5de9a39d7
 SHA512:
-  metadata.gz: 62fa904711056d3c7a10d395f93ace2012f63cc4ad84cd4e527c01fab1b9e092b53160d39d0f9cedd834cc079a2fbd24c6ce9cb21babb708fd7a093edd0b32f3
-  data.tar.gz: ada725ee7a06c66c9946101aa46ff02a28c105fd6c8ebaab7d822819d84a5a657dc8d23e1dfba529997f474024b6d64678695a2616ccea3ad07f664f01681edc
+  metadata.gz: 4cf93ae96552f60927f148fa187771b02b2cbff0717b66bbc94d58a69d84ce3aee0d23f1917e907bfd576a0721559641450103ebc491b3c534b867d4c2cc5c60
+  data.tar.gz: 20cff17b463ee4da7831614877394716447e4a71af0ed7104b80b12c4b33dc4653077f7bb27c04875539d80881a2535492e4567950d3256f0686084bcc56ab9d

data/lib/music_detector/multiple_linear_regression.rb

@@ -63,6 +63,7 @@ module MusicDetector
     end
 
     # @param [NArray] fvs array of input feature vector(s)
+    # @return [Array<Boolean>] estimated result; if the n-th element is true, the n-th feature vector might contain music
     def estimate(fvs)
       raise ArgumentError('the form of the input is differ from trained model') if @b.total - 1 != fvs.shape[0]
 
@@ -78,7 +79,10 @@ module MusicDetector
           x[1..(fvs.shape[0]), i] = fvs[true, i].flatten
         end
       end
-      x * @b
+      y = x * @b
+      y[y < 0] = false
+      y[y >= 0] = true
+      y.to_a
     end
 
     def export_to(file_path, config:)

data/lib/music_detector/version.rb

@@ -1,3 +1,3 @@
 module MusicDetector
-  VERSION = "0.1.2"
+  VERSION = "0.1.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: music_detector
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - sobataro