Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #save classes as ints
- #create map from class->index
- images, classes = loader.loadTrainingAndClasses()
- unique_classes = list(set(classes))
- int_unique_classes={unique_classes[i]:i for i in range(len(unique_classes))}
- #convert classes with map
- int_classes=[int_unique_classes[cl] for cl in classes]
- int_classes=numpy.array(int_classes)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement