NN_classifier module

class NN_classifier.NeuralClassifier(network_path, label_path)

Bases: object

Uses a neural network to predict attractiveness of a picture of a human.

Parameters:

• network_path – Path to the saved neural network.
• label_path – Path to the label text file.

predict(image)

Reads an image and returns a list of labels and scores in descending order.

Parameters:image – The image to score.