Loading deepchem/data/data_loader.py +6 −3 Original line number Diff line number Diff line Loading @@ -352,6 +352,7 @@ class ImageLoader(DataLoader): def featurize(self, input_files, labels=None, weights=None, read_img=True, in_memory=True): """Featurizes image files. Loading Loading @@ -404,13 +405,15 @@ class ImageLoader(DataLoader): else: X = [None] * len(image_files) if in_memory: return NumpyDataset(X, y=labels, ids=image_files) return NumpyDataset(X, y=labels, w=weights, ids=image_files) else: # from_numpy currently requires labels. Make dummy labels if labels is None: labels = np.zeros((len(image_files), 1)) return DiskDataset.from_numpy(X, labels, ids=image_files) if weights is None: weights = np.zeros((len(image_files), 1)) return DiskDataset.from_numpy(X, labels, w=weights, ids=image_files) @staticmethod def load_img(image_files): Loading Loading
deepchem/data/data_loader.py +6 −3 Original line number Diff line number Diff line Loading @@ -352,6 +352,7 @@ class ImageLoader(DataLoader): def featurize(self, input_files, labels=None, weights=None, read_img=True, in_memory=True): """Featurizes image files. Loading Loading @@ -404,13 +405,15 @@ class ImageLoader(DataLoader): else: X = [None] * len(image_files) if in_memory: return NumpyDataset(X, y=labels, ids=image_files) return NumpyDataset(X, y=labels, w=weights, ids=image_files) else: # from_numpy currently requires labels. Make dummy labels if labels is None: labels = np.zeros((len(image_files), 1)) return DiskDataset.from_numpy(X, labels, ids=image_files) if weights is None: weights = np.zeros((len(image_files), 1)) return DiskDataset.from_numpy(X, labels, w=weights, ids=image_files) @staticmethod def load_img(image_files): Loading
