TraceAssist (version 0.1.0)

CNN: Convolutional Neural Network (CNN) with two hidden layers

Description

Implement a CNN with two hidden layers and ReLU activation.

Usage

CNN(X,y,X_new,plot.figure = FALSE)

Arguments

X

A list of matrix-valued predictors.

y

Binary response variable.

X_new

A list of new matrices in the test data.

plot.figure

Option for plotting trajectory of accuracy over epochs.

Value

The returned object is a list of components.

prob - The predicted probabilities for the test data.

class - The estimated binary response for the test data.

history - The trajectory of classification accuracy over epochs.

acc - The classification accuracy on test data.