Exemple #1
0
def infer_from_processed(model_name, processed_file):
    model = load_model(model_name)
    print('infering from model "{}"'.format(model_name))
    predictions = model.predict_generator(
        generator=DataGenerator.__genfileforinference__(
            processed_file, constants.CONTEXT_WINDOW_SIZE,
            constants.SEQUENCE_SIZE),
        steps=DataGenerator.__inferencelen__(processed_file,
                                             constants.SEQUENCE_SIZE))

    print('postprocessing')
    predictions = np.reshape(
        predictions,
        (predictions.shape[0] * predictions.shape[1], predictions.shape[2]))
    predictions = np.rint(predictions)

    return predictions