Newer
Older
// Program 5: configure a model with metrics for accuracy and recall, then evaluate
dataset5.split(0.75)
dataset5.generateDataLoader()
model5.setLoss(crossEntropy)
model5.addMetrics(accuracy(0.9), recall(0.8))
model5.compile(linear(4, 256), applyLeakyReLU(0.01), linear(256, 3), applySoftmax(0.4))