kusari 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 1bd23a95575a6e870596ea70b119829c01cbeaf0
-  data.tar.gz: e52a64cf85a4993763df1121d9cd0b7adb53d6c2
+  metadata.gz: 74fdef5cc781f68d8fa0649852196e18da653bb4
+  data.tar.gz: 4527dd51cc6a9457232417b710d6dba6ee575a9c
 SHA512:
-  metadata.gz: 8ead23f301fa3745b1dc25e19b11f7d225595868cd365b1c067c17ae7dc393502cbaed79dcd63ae121ba15a352dbd0971c1ae3b55a34570258ef168ca589f0db
-  data.tar.gz: 7afdbd2f11989ddcdde5ae67d8bcd3d59156e7fe1a4965c870827c699c6e9a6ef3d8b67965604eb5e6c13e24f9c1e9b077a9d01a6ca8a78c66e9b8b80e7fec79
+  metadata.gz: 8e71daa28c77501ef7161c8216f1b077492396d4c62d7125e1ab1fd3eca77f2232fdcd7ce94727b3e95c0ffccb478d6de727dd0262b36d3d0883c3c654472a73
+  data.tar.gz: 6828744572889782705e67bf3f04c18f46dd7ad32d0014a9b67371a264799d7688a61c2ad55d9abdf26e09acf9fccf58260dc12d170d138ce4b9cd8a8f83362e

data/README.md

@@ -1,4 +1,4 @@
-# :link: Kusari
+# :link: Kusari [![Gem Version](https://badge.fury.io/rb/kusari.png)](https://badge.fury.io/rb/kusari)
 
 Japanese random sentence generator based on Markov chain.
 

data/lib/kusari.rb

@@ -4,7 +4,7 @@ require "kusari/version"
 module Kusari
   class Generator
     def initialize(gram=3, ipadic_path="./ipadic")
-      @generator = MarkovSentenceGenerator.new(gram, ipadic_path)
+      @generator = MarkovSentenceGenerator.new([2, gram].max, ipadic_path)
     end
 
     def add_string(string)

data/lib/kusari/markov_sentence_generator.rb

@@ -26,8 +26,8 @@ class MarkovSentenceGenerator
   def add(string)
     tokens = tokenize(string)
 
-    # if there are at least 4 tokens, we can create both of HEAD-started and TAIL-ended array of words
-    return if tokens.size < 4
+    # if there are at least N+1 tokens, we can create both of HEAD-started and TAIL-ended array of words
+    return if tokens.size < @gram + 1
 
     # update markov_table
     i = 0
@@ -49,14 +49,14 @@ class MarkovSentenceGenerator
 
     # sample one HEAD-started array and create initial sentence based on that
     sampled_array = head_arrays.sample
-    sentence = sampled_array[1] + sampled_array[2]
+    sentence = sampled_array[1..@gram-1].join
 
     # start Markov chain until getting the TAIL flag
     loop do
       # select all arrays which can chain their head word to current tail of the sentence
       chain_arrays = Array.new
       @markov_table.each do |markov_array|
-        if markov_array[0] == sampled_array[2]
+        if markov_array[0] == sampled_array[@gram-1]
           chain_arrays << markov_array
         end
       end
@@ -66,11 +66,11 @@ class MarkovSentenceGenerator
 
       # grow current sentence and check if it has the TAIL flag
       sampled_array = chain_arrays.sample
-      if sampled_array[2] == TAIL
-        sentence += sampled_array[1]
+      if sampled_array[sampled_array.length-1] == TAIL
+        sentence += sampled_array[1..@gram-2].join
         break
       else
-        concat_string = sampled_array[1] + sampled_array[2]
+        concat_string = sampled_array[1..@gram-1].join
         break if sentence.length + concat_string.length > limit
         sentence += concat_string
       end

data/lib/kusari/version.rb

@@ -1,3 +1,3 @@
 module Kusari
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: kusari
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - takuti
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2015-12-09 00:00:00.000000000 Z
+date: 2015-12-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: igo-ruby