Contents

module Tomoto
  class HLDA
    def self.new(tw: :one, min_cf: 0, min_df: 0, rm_top: 0, depth: 2, alpha: 0.1, eta: 0.01, gamma: 0.1, seed: nil)
      model = _new(to_tw(tw), depth, alpha, eta, gamma, seed || -1)
      model.instance_variable_set(:@min_cf, min_cf)
      model.instance_variable_set(:@min_df, min_df)
      model.instance_variable_set(:@rm_top, rm_top)
      init_params(model, binding)
    end

    def children_topics(topic_id)
      check_topic(topic_id)
      _children_topics(topic_id)
    end

    def level(topic_id)
      check_topic(topic_id)
      _live_topic?(topic_id) ? _level(topic_id) : -1
    end

    def live_topic?(topic_id)
      check_topic(topic_id)
      _live_topic?(topic_id)
    end

    def num_docs_of_topic(topic_id)
      check_topic(topic_id)
      _num_docs_of_topic(topic_id)
    end

    def parent_topic(topic_id)
      check_topic(topic_id)
      _live_topic?(topic_id) ? _parent_topic(topic_id) : -1
    end

    private

    def check_topic(topic_id)
      raise "topic_id must be < K" if topic_id >= k
      raise "train() should be called first" unless @prepared
    end

    def topics_info(summary, topic_word_top_n:)
      counts = count_by_topics

      nested_info = lambda do |k = 0, level = 0|
        words = topic_words(k, top_n: topic_word_top_n).keys.join(" ")
        summary << "| #{"  " * level}##{k} (#{counts[k]}) : #{words}"
        children_topics(k).sort.each do |c|
          nested_info.call(c, level + 1)
        end
      end
      nested_info.call
    end
  end
end

Version data entries

31 entries across 31 versions & 1 rubygems

Version	Path
tomoto-0.3.1	lib/tomoto/hlda.rb
tomoto-0.3.0-x86_64-linux	lib/tomoto/hlda.rb
tomoto-0.3.0-x86_64-darwin	lib/tomoto/hlda.rb
tomoto-0.3.0	lib/tomoto/hlda.rb
tomoto-0.2.3	lib/tomoto/hlda.rb
tomoto-0.2.2	lib/tomoto/hlda.rb
tomoto-0.2.1	lib/tomoto/hlda.rb
tomoto-0.2.0	lib/tomoto/hlda.rb
tomoto-0.1.4	lib/tomoto/hlda.rb
tomoto-0.1.3	lib/tomoto/hlda.rb
tomoto-0.1.2	lib/tomoto/hlda.rb