test.py

from cleanup_data import X_test, y_test
from model import model
from utils import getGenre
import numpy as np

NUMBER_VALUES = 1000


def formatPrediction(predictions, y_test):
    good = 0
    i = 0
    for prediction, y in zip(predictions, y_test):

        prediction_genre = getGenre(prediction)
        y_genre = getGenre(y)
        if prediction_genre == y_genre:
            good += 1
        if i % 50 == 0:
            print(
                f"{i} | prediction: {prediction_genre} | real result: {y_genre}")
        i += 1
    print(
        f"\ngood/total: {good}/{len(predictions)} - {round(100*good/len(predictions))}%")


model.load_weights('./models/bestmodel.h5')


predictions = model.predict(X_test[:NUMBER_VALUES])

formatPrediction(predictions, y_test[:NUMBER_VALUES])