Module: Chbs

Defined in:
lib/chbs.rb,
lib/chbs/version.rb

Defined Under Namespace

Classes: Generator

Constant Summary collapse

CORPORA_DIRECTORY =
File.expand_path('../../corpora/', __FILE__)
DEFAULT_CORPUS =
'tv-and-movies'
VERSION =
"0.0.9"

Class Method Summary collapse

Class Method Details

.included_corporaObject



9
10
11
# File 'lib/chbs.rb', line 9

def self.included_corpora
  Dir.glob(File.join(CORPORA_DIRECTORY, '*.json')).collect{|c| File.basename(c, '.json')}.sort
end

.load_corpus(corpus = DEFAULT_CORPUS) ⇒ Object



13
14
15
16
17
18
19
20
21
22
# File 'lib/chbs.rb', line 13

def self.load_corpus(corpus=DEFAULT_CORPUS)
  corpus ||= DEFAULT_CORPUS
  corpusfile = nil
  if corpus.include?('/')
    corpusfile = corpus
  else
    corpusfile = File.join(CORPORA_DIRECTORY, "#{corpus}.json")
  end
  JSON.parse(File.read(corpusfile))
end