public class ListenerEvaluations extends Object
Usage example - does classification (Evaluation
) evaluation on
the training set (as training proceeds) and also Evaluation/ROCMultiClass evaluation on the test/validation set.
Assumes that the output predictions are called "softmax" and the first DataSet/MultiDataSet labels are those corresponding
to the "softmax" node
ListenerEvaluations.builder()
//trainEvaluations: on the training data (in-line, as training proceeds through the epoch)
.trainEvaluation("softmax", 0, new Evaluation(), new ROCMultiClass())
//validationEvaluation: on the test/validation data, at the end of each epoch
.validationEvaluation("softmax", 0, new Evaluation(), new ROCMultiClass())
.build();
Modifier and Type | Class and Description |
---|---|
static class |
ListenerEvaluations.Builder |
Constructor and Description |
---|
ListenerEvaluations(Map<String,List<IEvaluation>> trainEvaluations,
Map<String,Integer> trainEvaluationLabels,
Map<String,List<IEvaluation>> validationEvaluations,
Map<String,Integer> validationEvaluationLabels) |
Modifier and Type | Method and Description |
---|---|
static ListenerEvaluations.Builder |
builder() |
boolean |
isEmpty() |
ListenerVariables |
requiredVariables()
Get the required variables for these evaluations
|
Map<String,Integer> |
trainEvaluationLabels()
Get the label indices for the requested training evaluations
|
Map<String,List<IEvaluation>> |
trainEvaluations()
Get the requested training evaluations
|
Map<String,Integer> |
validationEvaluationLabels()
Get the label indices for the requested validation evaluations
|
Map<String,List<IEvaluation>> |
validationEvaluations()
Get the requested validation evaluations
|
public static ListenerEvaluations.Builder builder()
public Map<String,List<IEvaluation>> trainEvaluations()
public Map<String,Integer> trainEvaluationLabels()
public Map<String,List<IEvaluation>> validationEvaluations()
public Map<String,Integer> validationEvaluationLabels()
public ListenerVariables requiredVariables()
public boolean isEmpty()
Copyright © 2020. All rights reserved.