tiny-classifier 2.0 → 2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: ff0e4cd3aafc37e4ba99b782d3b7816c6bbb5f9f
-  data.tar.gz: d9b99162fe1711f3f7cdaaf3791f1df9d75df14f
+  metadata.gz: 7eb179e8a3a17711921f6bd0af6132909ace2f90
+  data.tar.gz: 2f029f934ddb3f5277980228800fa24093491bab
 SHA512:
-  metadata.gz: 933a1773de70ee773863a643556ec4dfedff6433ee3385b4ce8f3283efa1959e25027fae1712f410c39d70f72b74a06b231299d5041a12276afe8c771f948047
-  data.tar.gz: 93602fc9af2bda18c52beaefd980b912aa438b95526c549d965fde192c3f3061fbb1ebffaf13de221ec92ae57e7cf25b06adb3c0c83c90f16520aa3208037cab
+  metadata.gz: 9ecc0831b783aed6891e09fe73260f0eef63f021e450bf4db89cea267b4797cffade8f74b05530a058646d77ca1265a24ea7287e6754e132ec1e724d0ae21611
+  data.tar.gz: aa5e0822ebcbc19b5497f1596be44b3b01e0fe184903d8b111fa9eae635f5bfd8c94ed34bda5a98658514cf10d185806abbd0eb12a124beba92943750bf107a0

data/README.md

@@ -67,7 +67,7 @@ positive
 
 ### Common
 
-`-l`, `--categories=CATEGORIES` (required)
+`-c`, `--categories=CATEGORIES` (required)
 :  A comman-separated list of categories. You should use only alphabetic characters. (Non-alphabetical characters will cause problems.)
 
 `-d`, `--data-dir=PATH` (optional)

data/lib/tiny-classifier/base.rb

@@ -25,6 +25,7 @@ module TinyClassifier
     def initialize
       @tokenizer = Tokenizer.new
       @data_dir = Dir.pwd
+      @verbose = false
     end
 
     def parse_command_line_options(command_line_options)
@@ -51,6 +52,7 @@ module TinyClassifier
       parser.on("-c CATEGORIES", "--categories=CATEGORIES",
                 "List of categories (comma-separated)") do |categories|
         @categories = normalize_categories(categories)
+        log("categories: #{@categories}")
       end
 
       parser.on("-t TOKENIZER", "--tokenizer=TOKENIZER",
@@ -58,6 +60,11 @@ module TinyClassifier
         @tokenizer.type = tokenizer
       end
 
+      parser.on("-v", "--verbose",
+                "Output internal information (for debugging)") do |verbose|
+        @verbose = verbose
+      end
+
       parser
     end
 
@@ -107,12 +114,23 @@ module TinyClassifier
 
     def prepare_input
       unless File.pipe?(STDIN)
-        STDERR.puts("Error: No effective input. You need to give any input via the STDIN.")
+        error("Error: No effective input. You need to give any input via the STDIN.")
         exit(false)
       end
       @input = $stdin.readlines.join(" ")
-      @tokenizer.tokenize(@input)
+      @input = @tokenizer.tokenize(@input)
+      log("tokenizer: #{@tokenizer.type}")
       @input.strip!
+      log("input: #{@input}")
+      @input
+    end
+
+    def error(message)
+      STDERR.puts(message)
+    end
+
+    def log(message)
+      STDERR.puts(message) if @verbose
     end
   end
 end

data/lib/tiny-classifier/classifier.rb

@@ -30,7 +30,7 @@ module TinyClassifier
 
     def run
       if input.empty?
-        STDERR.puts("Error: No effective input.")
+        error("Error: No effective input.")
         false
       else
         category = classifier.classify(input)

data/lib/tiny-classifier/retrainer.rb

@@ -29,7 +29,7 @@ module TinyClassifier
 
     def run(params)
       if input.empty?
-        STDERR.puts("Error: No effective input.")
+        error("Error: No effective input.")
         false
       else
         @category = params[:wrong]

data/lib/tiny-classifier/tokenizer.rb

@@ -27,7 +27,7 @@ module TinyClassifier
     end
 
     def tokenize(input)
-      case @tokenizer.to_s.downcase.to_sym
+      case @type.to_s.downcase.to_sym
       when :mecab
         tokenize_by_mecab(input)
       else

data/lib/tiny-classifier/trainer.rb

@@ -35,7 +35,7 @@ module TinyClassifier
       @category = params[:category]
       prepare_category
       if input.empty?
-        STDERR.puts("Error: No effective input.")
+        error("Error: No effective input.")
         false
       else
         classifier.send("train_#{@category}", input)
@@ -47,19 +47,19 @@ module TinyClassifier
     private
     def prepare_category
       unless @category
-        STDERR.puts("Error: You need to specify the category for the input.")
+        error("Error: You need to specify the category for the input.")
         exit(false)
       end
 
       @category = @category.downcase.strip
 
       if @category.empty?
-        STDERR.puts("Error: You need to specify the category for the input.")
+        error("Error: You need to specify the category for the input.")
         exit(false)
       end
 
       unless @categories.include?(@category.capitalize)
-        STDERR.puts("Error: You need to specify one of valid categories: #{@categories.join(', ')}")
+        error("Error: You need to specify one of valid categories: #{@categories.join(', ')}")
         exit(false)
       end
     end

data/lib/tiny-classifier/untrainer.rb

@@ -21,7 +21,7 @@ module TinyClassifier
       @category = params[:category]
       prepare_category
       if input.empty?
-        STDERR.puts("Error: No effective input.")
+        error("Error: No effective input.")
         false
       else
         classifier.send("untrain_#{@category}", input)

data/tiny-classifier.gemspec

@@ -21,7 +21,7 @@ $LOAD_PATH.unshift(File.join(File.dirname(__FILE__), "lib"))
 
 Gem::Specification.new do |spec|
   spec.name = "tiny-classifier"
-  spec.version = "2.0"
+  spec.version = "2.1"
   spec.homepage = "https://github.com/piroor/tiny-classifier"
   spec.authors = ["YUKI \"Piro\" Hiroshi"]
   spec.email = ["piro.outsider.reflex@gmail.com"]

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tiny-classifier
 version: !ruby/object:Gem::Version
-  version: '2.0'
+  version: '2.1'
 platform: ruby
 authors:
 - YUKI "Piro" Hiroshi