src.model.classification_model

Classes

ClassificationModel(**kwargs)

class src.model.classification_model.ClassificationModel(**kwargs)
Author:

Alberto M. Esmoris Pena

Abstract class providing some baseline methods for classification models.

static extract_model_args(spec)

Extract the arguments to initialize/instantiate a ClassificationModel from a key-word specification.

Parameters:

spec – The key-word specification containing the arguments.

Returns:

The arguments to initialize/instantiate a ClassificationModel.

__init__(**kwargs)

Initialization for any instance of type ClassificationModel.

overwrite_pretrained_model(spec)

See src.model.model.Model.overwrite_pretrained_model().

autoval(y, yhat, info=True)

Auto validation during training for classification models.

Parameters:

  • yhat – The predicted classes.

  • y – The expected classes.

  • info – True to log an info message with the auto validation, False otherwise.

Returns:

The results of the auto validation.

Return type:

np.ndarray

on_training_finished(X, y, yhat=None)

See model.Model.on_training_finished().

static autoval_metrics_from_names(names)

See src.eval.classification_evaluator.ClassificationEvaluator.metrics_from_names() .