keraspredictionfunction

keraspredictionfunction ¶

module for keras prediction function

Classes¶

KerasPredictionFunction ¶

Bases: PredictionFunction

Prediction function for keras models

Functions¶

predict ¶

predict(model, data_x)

uses a keras model to predict the data

Source code in niceml/dlframeworks/keras/predictionfunctions/keraspredictionfunction.py

def predict(self, model, data_x) -> Any:
    """uses a keras model to predict the data"""
    pred = model.predict_step(data_x).numpy()
    return pred