53a93d8af5d041f6c5dfdef2fb763f8419e3023d,matchzoo/datasets/snli/load_data.py,,load_data,#Any#Any#Any#,14

Before Change

        raise ValueError(f"{stage} is not a valid stage."
                         f"Must be one of `train`, `dev`, and `test`.")

    data_root = _download_data()
    file_path = data_root.joinpath(f"snli_1.0_{stage}.txt")
    data_pack = _read_data(file_path)

    if task == "ranking":
        task = matchzoo.tasks.Ranking()
    if task == "classification":
        task = matchzoo.tasks.Classification()

    if isinstance(task, matchzoo.tasks.Ranking):
        if target_label not in ["entailment", "contradiction", "neutral", "-"]:
            raise ValueError
        binary = (data_pack.relation["label"] == target_label).astype(float)
        data_pack.relation["label"] = binary
        return data_pack
    elif isinstance(task, matchzoo.tasks.Classification):
        classes = ["entailment", "contradiction", "neutral", "-"]

After Change

        data_pack.relation["label"] = binary
        return data_pack
    elif isinstance(task, matchzoo.tasks.Classification):
        classes = ["entailment", "contradiction", "neutral", "-"]
        label = data_pack.relation["label"].apply(classes.index)
        data_pack.relation["label"] = label
        data_pack.one_hot_encode_label(num_classes=4, inplace=True)

Instances