Unverified Commit b49cd095 authored by Xiangkun Hu's avatar Xiangkun Hu Committed by GitHub
Browse files

[Bugfix] Fix feature type of 'ENZYMES' in TUDataset (#2800)



* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* Hide implementations that may cause confusions to the user

* fix number of classes in Amazon Cobuy datasets

* add four papers

* fix feature type of TU dataset

* fix readme

* remove commented codes
Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
parent 1f4c0b71
...@@ -324,7 +324,7 @@ class TUDataset(DGLBuiltinDataset): ...@@ -324,7 +324,7 @@ class TUDataset(DGLBuiltinDataset):
for filename, field_name in self.attr_dict.items(): for filename, field_name in self.attr_dict.items():
try: try:
data = loadtxt(self._file_path(filename), data = loadtxt(self._file_path(filename),
delimiter=',').astype(int) delimiter=',')
if 'label' in filename: if 'label' in filename:
data = F.tensor(self._idx_from_zero(data)) data = F.tensor(self._idx_from_zero(data))
else: else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment