cnut1648/

mnli_resampled_as_mednli

This dataset inherits from mnli, and the only changes is to downsample to the same size as mednli. It is created via with open("/path/to/mednli/mli_train_v1.jsonl") as f: data = [json.loads(line) for line in f.readlines()]

raw_datasets = load_dataset("glue", "mnli") raw_datasets["train"] = raw_datasets["train"].shuffle(seed=42).select(range(len(data))) raw_datasets.push_to_hub("cnut1648/mnli_resampled_as_mednli")