Hive_ML.training.model_trainer module#

Hive_ML.training.model_trainer.model_fit_and_predict(model, x_train, y_train, x_val)[source]#

Function to fit a SKLearn with the given training features and training labels, returning the predicted the labels for the validation features.

Parameters:

  • model (ClassifierMixin) – a SKlearn model.

  • x_train (ndarray) – training features.

  • y_train (ndarray) – training labels.

  • x_val (ndarray) – validation features.

Return type:

ndarray

Returns:

predicted validation labels.