Get Classification Metrics.R

get_metrics <- function(model, test_data, truth, table = FALSE){
  preds <- predict(model, test_data);
  table_ <- table(truth, preds);
  acc_ <- sum(diag(table_))/sum(table_);
  rec_ <- table_[2,2]/sum(table_[,2]);
  prec_ <- table_[2,2]/sum(table_[2,]);
  f_ <- 2*rec_*prec_/(rec_+prec_)
  if (table == FALSE){
    cat('Accuracy:', acc_, '\nRecall:', rec_, '\nPrecision:', prec_, '\nF-Score:', f_);
  } else {
    return(table_)
  }
  
    # return(c(acc_, rec_, prec_, f_));
}