Commit cf4a3c63 authored by Alessia Marcolini's avatar Alessia Marcolini
Browse files

Use labels as integers; TODO: label encoder

parent 65b19d15
......@@ -29,7 +29,7 @@ class TileDataset(BaseDataset):
metadata_tiles = metadata_tiles[metadata_tiles['mode'] == mode]
self.filenames = metadata_tiles['tile_filename'].values
self.labels = metadata_tiles['label'].astype('str').values
self.labels = metadata_tiles['label'].values
self.patients = metadata_tiles['patient'].values
def __getitem__(self, i):
......
Supports Markdown
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