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.

Module contents