stopping_criteria {autoEnsemble}

Stopping Criteria for Ending the Search

Description

Defines criteria for ending the optimization search

Usage

stopping_criteria(
  df,
  round,
  stop,
  min_improvement,
  stop_rounds = 3,
  reset_stop_rounds = TRUE,
  stop_metric = "auc"
)

Arguments

df

data.frame. includes the metrics of ensemblem model performance

round

integer. the current round of optimization

stop

integer. current round of stopping penalty

min_improvement

numeric. specifies the minimum improvement in model evaluation metric to qualify further optimization search.

stop_rounds

integer. number of stoping rounds, in case the model stops improving

reset_stop_rounds

logical. if TRUE, everytime the model improves the stopping rounds penalty is resets to 0.

stop_metric

character. model stopping metric. the default is "auc", but "aucpr" and "mcc" are also available.

Value

a matrix of F-Measures for different thresholds or the highest F-Measure value

Author(s)

E. F. Haghish