proposal-classifier 0.3.0 → 0.4.0

Sign up to get free protection for your applications and to get access to all the features.
checksums.yaml CHANGED
@@ -1,7 +1,7 @@
1
1
  ---
2
2
  SHA256:
3
- metadata.gz: 825ce2cda03f9c46f92acff84ca7fdb7e24e14f57996ed3737843b8e01723330
4
- data.tar.gz: 817a1dd9f68cb1ebefe1ef2d68b5b3f1bf38ff72cc5bcf9624ea71a418d2801f
3
+ metadata.gz: 893a46b5e6065062c274ee5246157547908099fd6ee57f151755162faef368b3
4
+ data.tar.gz: 4a679e68b111835288a15919e5908350170ea5b97f3d9fa35632b3e95d097ef1
5
5
  SHA512:
6
- metadata.gz: 139278c11a9642fe1499cbac5a9de1bf1bd219321def8216fd07954d681a808ee71c92cbbd2e0b07948dd0d34976f05bc9a6a9f51cda691621acaa33e98198fd
7
- data.tar.gz: 64d508d35f6b69e1b29a6d56e77f19db257b973a04f3b6d38e7f7c8c8b2072961b5c83660eda2fbc78656b291a41acb7bc3b2d61a2e95f815dead4ce4061bb44
6
+ metadata.gz: 429fbbdb58a211a5dd38e2a6fc18195fc4acac91cdc01b20b0450d1738d40d50d2cf9c75606382d9eebc7cb5ae6a176452469ec687a83db2bdd20592ae07d306
7
+ data.tar.gz: 2ac79390660c14d39bb2f9988ffe29fe805f019022a3075f77306245044676b43f9229b9a7b68c5aedcca119d37168f414e17a6b838ee99b3c9191884d01b4bb
@@ -2,6 +2,6 @@
2
2
 
3
3
  module Proposal
4
4
  module Classifier
5
- VERSION = "0.3.0"
5
+ VERSION = "0.4.0"
6
6
  end
7
7
  end
@@ -9,12 +9,12 @@ require_relative "classifier/version"
9
9
  module Proposal
10
10
  module Classifier
11
11
  class Model
12
- def self.predict_onx(input_text)
12
+ def self.predict_onx(input_text, onnx_model_path)
13
13
  nlp = Spacy::Language.new("pt_core_news_lg")
14
14
  doc = nlp.read(input_text)
15
15
  vector = doc.vector
16
16
  scaled_array = (vector - vector.min) / (vector.max - vector.min) * (1 - 0) + 0
17
- model = OnnxRuntime::InferenceSession.new("./exported_model.onnx")
17
+ model = OnnxRuntime::InferenceSession.new(onnx_model_path)
18
18
  label_name = model.outputs()[0][:name]
19
19
  vector_fim_reshaped = scaled_array.reshape(1, -1)
20
20
  pred_onx = model.run([label_name], {X: vector_fim_reshaped})[0]
metadata CHANGED
@@ -1,7 +1,7 @@
1
1
  --- !ruby/object:Gem::Specification
2
2
  name: proposal-classifier
3
3
  version: !ruby/object:Gem::Version
4
- version: 0.3.0
4
+ version: 0.4.0
5
5
  platform: ruby
6
6
  authors:
7
7
  - chaydson