store_classification_results: Store results from keras tf classification training and...

Description Usage Arguments Value

View source: R/keras_classification.R

Description

This function extracts information from the keras model generated by training or prediction and stores them in a data.frame. By calling 'classification_metrics' various metrics for classification performance are calculated for each class.

Usage

1
2
3
4
5
6
7
store_classification_results(
  hist,
  prediction_table,
  n_classes,
  confusion_matrix,
  train_data
)

Arguments

hist

the keras history object

prediction_table

the data.frame comparing predictions and true values

n_classes

the number of classes for classification

confusion_matrix

the confusion matrix generated from 'prediction_table'

train_data

the training set data.frame

Value

A data frame with one row per keras run and class


RJ333/phyloseq2ML documentation built on June 2, 2020, 9:25 p.m.