Unverified Commit 4a87c36f authored by Suzukazole's avatar Suzukazole
Browse files

changed feature field

parent 0c4020ca
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -60,7 +60,7 @@ class _USPTOLoader(_MolnetLoader):

    loader = dc.data.CSVLoader(
        tasks=self.tasks,
        feature_field=["src", "tgt"],
        feature_field=["reactions"],
        featurizer=self.featurizer)

    return loader.create_dataset(dataset_file, shard_size=8192)