nodes_dtype,):assertisinstance(dataset, (Cora, CiteSeer))ifnodes_dtypeisNone: nodes_dtype = dataset._NODES_DTYPE dataset.download()# expected_files should be in this ordercites, content = [dataset._resolve_path(name)fornameindataset.expected_files] feature_names = ["w_{}".format(ii)foriii...
(train_data, shuffle=True, batch_size=train_batch_size) train_loss = losses.CosineSimilarityLoss(model=model) dev_data = SentencesDataset(examples=sts_reader.get_examples('sts-dev.csv'), model=model) dev_dataloader = DataLoader(dev_data, shuffle=False, batch_size=train_batch_size) evaluator...