matchzoo.datasets.snli
¶
Submodules¶
Package Contents¶
-
matchzoo.datasets.snli.
load_data
(stage:str='train', task:str='classification', target_label:str='entailment', return_classes:bool=False) → typing.Union[matchzoo.DataPack, tuple]¶ Load SNLI data.
Parameters: - 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)
- return_classes – True to return classes for classification task, False otherwise.
Returns: A DataPack unless task is classificiation and return_classes is True: a tuple of (DataPack, classes) in that case.