Advertisement
hdbondt

Untitled

Nov 27th, 2015
72
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.34 KB | None | 0 0
  1. #save classes as ints
  2. #create map from class->index
  3. images, classes = loader.loadTrainingAndClasses()
  4.  
  5. unique_classes = list(set(classes))
  6. int_unique_classes={unique_classes[i]:i for i in range(len(unique_classes))}
  7.  
  8. #convert classes with map
  9. int_classes=[int_unique_classes[cl] for cl in classes]
  10. int_classes=numpy.array(int_classes)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement