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)¶