quick_anomaly_detector.models.TrainClassificationNN

class quick_anomaly_detector.models.TrainClassificationNN(lr=0.001, num_epochs=1000, patience=10)
__init__(lr=0.001, num_epochs=1000, patience=10)

Methods

__init__([lr, num_epochs, patience])

predict(X[, y_val])

train(X_train, X_valid, y_train, y_val)