matchzoo.datasets.snli package¶
Submodules¶
matchzoo.datasets.snli.load_data module¶
SNLI data loader.
-
matchzoo.datasets.snli.load_data.
load_data
(stage='train', task='classification', target_label='entailment')¶ Load SNLI data.
参数: - stage -- One of train, dev, and test. (default: train)
- task -- Could be one of ranking, classification or a
matchzoo.engine.BaseTask
instance. (default: ranking) - target_label -- If ranking, chose one of entailment, contradiction, neutral, and - as the positive label. (default: entailment)
返回: A DataPack if ranking, a tuple of (DataPack, classes) if classification.