music_detector 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4ba71bbdd7681ddbf73efa730ca882654a97e172
-  data.tar.gz: 688ab862011e79aee2fd7198aa5aa49ade15b8b0
+  metadata.gz: 9d4347bc2aa7368840aeacbfea0d5f8988198d96
+  data.tar.gz: b91ba65982e25f962fe90da3714ce9a55f145c0b
 SHA512:
-  metadata.gz: 664016744f6a149723cc517f93d9371dbda81ba6f7d4e0923d5f45a43282482a11444ee51f69586fa7552c840515a0521cb4bc759e4200ca79ba1f6bd5277dc9
-  data.tar.gz: af52b72b02ed0dd5244a632b8bf4978b972b32b7194281aab937e010840f225621c783bbe61da874f264ab550f6a39bce28e7e6b65eb2795de0c290c4bfee932
+  metadata.gz: 62fa904711056d3c7a10d395f93ace2012f63cc4ad84cd4e527c01fab1b9e092b53160d39d0f9cedd834cc079a2fbd24c6ce9cb21babb708fd7a093edd0b32f3
+  data.tar.gz: ada725ee7a06c66c9946101aa46ff02a28c105fd6c8ebaab7d822819d84a5a657dc8d23e1dfba529997f474024b6d64678695a2616ccea3ad07f664f01681edc

data/.gitignore

@@ -10,3 +10,4 @@
 
 # rspec failure tracking
 .rspec_status
+*.gem

data/lib/music_detector/multiple_linear_regression.rb

@@ -62,14 +62,21 @@ module MusicDetector
       model
     end
 
-    # @param [NArray]
-    def estimate(input)
-      raise RuntimeError('the form of the input is differ from trained model') if @b.total - 1 != input.shape[0]
-
-      x = NMatrix.float(input.shape[0] + 1, input.shape[1])
-      (input.shape[1]).times do |i|
-        x[0, i] = 1.0
-        x[1..(input.shape[0]), i] = input[true, i].flatten
+    # @param [NArray] fvs array of input feature vector(s)
+    def estimate(fvs)
+      raise ArgumentError('the form of the input is differ from trained model') if @b.total - 1 != fvs.shape[0]
+
+      x = nil
+      if fvs.shape.length == 1
+        x = NMatrix.float(fvs.shape[0] + 1, 1)
+        x[0, 0] = 1.0
+        x[1..(fvs.shape[0]), 0] = fvs
+      else
+        x = NMatrix.float(fvs.shape[0] + 1, fvs.shape[1])
+        (fvs.shape[1]).times do |i|
+          x[0, i] = 1.0
+          x[1..(fvs.shape[0]), i] = fvs[true, i].flatten
+        end
       end
       x * @b
     end

data/lib/music_detector/version.rb

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

data/music_detector.gemspec

@@ -11,7 +11,7 @@ Gem::Specification.new do |spec|
 
   spec.summary       = %q{music detector}
   spec.description   = %q{detects music on wave (.wav) file}
-  spec.homepage      = "https://github.com/sobataro/ruby-music-detector"
+  spec.homepage      = "https://github.com/sobataro/ruby-music_detector"
   spec.license       = "MIT"
 
   # Prevent pushing this gem to RubyGems.org. To allow pushes either set the 'allowed_push_host'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: music_detector
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - sobataro
@@ -123,7 +123,7 @@ files:
 - lib/music_detector/multiple_linear_regression.rb
 - lib/music_detector/version.rb
 - music_detector.gemspec
-homepage: https://github.com/sobataro/ruby-music-detector
+homepage: https://github.com/sobataro/ruby-music_detector
 licenses:
 - MIT
 metadata: {}