love_is 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: c47bfbb06c55765459c1f9080d35bf098b855233
+  data.tar.gz: 3aa2c252c09af30c32125841ca39fb528e3c6c84
+SHA512:
+  metadata.gz: 12271194b7454d53fa39ea62e7ad3eefeecce2c49fb84d3cf341c52c45176336c857f03b395796f6d5a643c53cc5a4a599191b2417b70ffd1043eb051ca6e67b
+  data.tar.gz: 3d2adc3c9478b84b5409b56ecd3992e3ba2d19882583c0297d32f00d1b5d02b0ba4415381441d27099202fa94d1a9f5e8473e233d4b31016b2e95061ecf5053c

data/lib/love_is/generator.rb

@@ -0,0 +1,32 @@
+class LoveIs::Generator
+  include LoveIs::Parser
+  attr_reader :ngram, :chain, :tree
+
+  def initialize(ngram = 3)
+    @tree = LoveIs::Node.new(nil, 1, [])
+    @ngram = ngram
+    @chain = []
+  end
+
+  def generate(chain_length = 1000)
+    @chain = []
+    initial_values
+    (chain_length - ngram).times do
+      break unless word = next_value
+      @chain << word
+    end
+    chain[1..-1].join(' ')
+  end
+
+  def initial_values
+    ngram.times.inject(@tree.child('*')) do |node|
+      @chain << node.value
+      node.next
+    end
+  end
+
+  def next_value
+    node = @tree[*chain[(-ngram + 1)..-1]]
+    node ? node.next.value : nil
+  end
+end

data/lib/love_is/node.rb

@@ -0,0 +1,62 @@
+class LoveIs::Node
+  attr_reader :value
+  attr_accessor :probability, :childs, :weight
+
+  def initialize(value = nil, weight = 1, childs = [])
+    @value = value
+    @weight = weight
+    set_childs(childs)
+  end
+
+  def set_childs(childs)
+    @childs = childs
+    unless @childs.all? { |x| x.probability }
+      probabilities
+    end
+  end
+
+  def probabilities
+    normalize_weights
+    t = 0.0
+    @childs.map! { |x| t += x.weight; x.probability = t; x }
+    @childs.each(&:probabilities) if childs && childs.any?
+  end
+
+  def normalize_weights
+    sum = weights.inject(&:+).to_f
+    @childs.map! { |x| x.weight /= sum; x }
+  end
+
+  def child(key)
+    childs.find { |x| x.value == key }
+  end
+
+  def [](*keys)
+    keys.inject(self) { |r, x| break(nil) unless r; r.child(x) }
+  end
+
+  def []=(*keys, weight)
+    keys.inject(self) do |r, x|
+      r.childs << LoveIs::Node.new(x, 0, []) unless r.child(x)
+      r.child(x).weight = weight
+      r.child(x)
+    end
+  end
+
+  def inc(*keys)
+    keys.inject(self) do |r, x|
+      r.childs << LoveIs::Node.new(x, 0, []) unless r.child(x)
+      r.child(x).weight += 1
+      r.child(x)
+    end
+  end
+
+  def weights
+    childs.map { |x| x.weight }
+  end
+
+  def next
+    random_value = Random.new.rand
+    childs.find { |x| random_value - x.probability <= 0 }
+  end
+end

data/lib/love_is/parser.rb

@@ -0,0 +1,20 @@
+module LoveIs::Parser
+  SENTENCE_BEGIN = '*'
+
+  def parse_file(filename)
+    text = File.read(File.expand_path(filename))
+    parse(text, @ngram)
+  end
+
+  def parse(text, length)
+    text.split("\n").map { |x| sentence2array(x) }.each do |sentence|
+      sentence.each_cons(length) { |sequence| @tree.inc(*sequence) }
+    end
+    @tree.probabilities
+    self
+  end
+
+  def sentence2array(sentence)
+    sentence.split(/\s/).unshift(SENTENCE_BEGIN)
+  end
+end

data/lib/love_is.rb

@@ -0,0 +1,8 @@
+#!/usr/bin/env ruby
+
+module LoveIs
+end
+
+require_relative 'love_is/node'
+require_relative 'love_is/parser'
+require_relative 'love_is/generator'

metadata

@@ -0,0 +1,48 @@
+--- !ruby/object:Gem::Specification
+name: love_is
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Sergey Smagin
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-12-07 00:00:00.000000000 Z
+dependencies: []
+description: Markov chains generator.
+email: smaginsergey1310@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/love_is.rb
+- lib/love_is/generator.rb
+- lib/love_is/node.rb
+- lib/love_is/parser.rb
+homepage: https://github.com/s-mage/love_is
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.3
+signing_key: 
+specification_version: 4
+summary: Love Is ...
+test_files: []
+has_rdoc: