Skip to content
Snippets Groups Projects
programme.maj 421 B
Newer Older
// Program 5: configure a model with metrics for accuracy and recall, then evaluate
Malvin Chevallier's avatar
Malvin Chevallier committed
newModel model5
configure {
Malvin Chevallier's avatar
Malvin Chevallier committed
    Dataset dataset5 = loadData("test.csv")
    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))
    model5.evaluate(dataset5)
}