beam.lua

--
-- Created by IntelliJ IDEA.
-- User: user
-- Date: 7/1/2016
-- Time: 8:47 PM
-- To change this template use File | Settings | File Templates.
--


require 'neuralconvo'
require 'util.Tester'
local tokenizer = require "tokenizer"
local list = require "pl.List"
require 'nn'
local WordSplitLMMinibatchLoader = require 'util.WordSplitLMMinibatchLoader'


local options = {}

if loader == nil then
    cmd = torch.CmdLine()
    cmd:text('Options:')
    cmd:option('--cuda', false, 'use CUDA. Training must be done on CUDA')
    cmd:option('--debug', false, 'show debug info')
    cmd:option('--dataset', "model.t7", 'show debug info')
    cmd:option('--vocablocation', "data/opensubssmall/vocabwords.t7", 'show debug info')
    cmd:text()
    options = cmd:parse(arg)


    -- Enabled CUDA
    if options.cuda then
        require 'cutorch'
        require 'cunn'
    end

    -- Data
    loader = WordSplitLMMinibatchLoader.createFromJustVocab(options.vocablocation)

end

if model == nil then
    print("-- Loading model")
    model = torch.load("data/"..options.dataset)
end



function say(text)
    print(getResponseBeam(text,loader,model,options.debug,5))
end


repeat
    io.write("Ask: ")
    io.flush()
    answer=io.read()

    io.write(say(answer))

until answer=="end"