tatk.e2e.rnn_rollout.deal_or_not package

Submodules

tatk.e2e.rnn_rollout.deal_or_not.model module

class tatk.e2e.rnn_rollout.deal_or_not.model.DealornotAgent(name, args, sel_args, train=False, diverse=False, max_total_len=100, model_url='https://tatk-data.s3-ap-northeast-1.amazonaws.com/rnnrollout_dealornot.zip')

Bases: tatk.e2e.rnn_rollout.rnnrollout.RNNRolloutAgent

The Rnn Rollout model for DealorNot dataset.

__init__(name, args, sel_args, train=False, diverse=False, max_total_len=100, model_url='https://tatk-data.s3-ap-northeast-1.amazonaws.com/rnnrollout_dealornot.zip')

Constructor of RNNRollout model.

auto_download()

Automatically download the pretrained model and necessary data.

tatk.e2e.rnn_rollout.deal_or_not.model.get_context_generator(context_file)