def add_cmdline_args(argparser):
"""
Add command-line arguments specifically for this agent.
"""
WizardTransformerRankerAgent.add_cmdline_args(argparser)
parser = argparser.add_argument_group('WizardRetrievalInteractive Arguments')
parser.add_argument(
'--retriever-model-file',
type=str,
default='models:wikipedia_full/tfidf_retriever/model',
)
parser.add_argument(
'--responder-model-file',
type=str,
default='models:wizard_of_wikipedia/full_dialogue_retrieval_model/model',
)
parser.add_argument(
'--get-unique',
type='bool',
default=True,
help='get unique responses from the bot',
)
parser.add_argument(
'--num-retrieved',
type=int,
default=7,
help='how many passages to retrieve for each' 'category',
)
parser.add_argument('--debug', type='bool', default=False)
return parser
of code changes.
Results on seen test set:
Hits@1/100: 86.7
Results on unseen test set (run with flag
`-t wizard_of_wikipedia:WizardDialogKnowledge:topic_split`):
Hits@1/100: 68.96
"""
if __name__ == '__main__':
parser = ParlaiParser(add_model_args=True)
parser.add_argument('-n', '--num-examples', default=100000000)
parser.add_argument('-d', '--display-examples', type='bool', default=False)
parser.add_argument('-ltim', '--log-every-n-secs', type=float, default=2)
WizardTransformerRankerAgent.add_cmdline_args(parser, partial_opt=None)
parser.set_params(
task='wizard_of_wikipedia',
model='projects:wizard_of_wikipedia:wizard_transformer_ranker',
model_file=
'models:wizard_of_wikipedia/full_dialogue_retrieval_model/model',
datatype='test',
n_heads=6,
ffn_size=1200,
embeddings_scale=False,
delimiter=' __SOC__ ',
n_positions=1000,
legacy=True,
)
opt = parser.parse_args()
of code changes.
Results on seen test set:
Hits@1/100: 86.7
Results on unseen test set (run with flag
`-t wizard_of_wikipedia:WizardDialogKnowledge:topic_split`):
Hits@1/100: 68.96
"""
if __name__ == '__main__':
parser = ParlaiParser(add_model_args=True)
parser.add_argument('-n', '--num-examples', default=100000000)
parser.add_argument('-d', '--display-examples', type='bool', default=False)
parser.add_argument('-ltim', '--log-every-n-secs', type=float, default=2)
WizardTransformerRankerAgent.add_cmdline_args(parser)
parser.set_defaults(
task='wizard_of_wikipedia',
model='projects:wizard_of_wikipedia:wizard_transformer_ranker',
model_file='models:wizard_of_wikipedia/retrieval_models/model',
datatype='test',
n_heads=6,
ffn_size=1200,
embeddings_scale=False,
delimiter=' __SOC__ ',
n_positions=1000,
legacy=True)
opt = parser.parse_args()
download(opt) # download pretrained retrieval model
