Attend the Women in ML Symposium on December 7 Register now

tf_privacy.single_layer_softmax_classifier

Stay organized with collections Save and categorize content based on your preferences.

Trains a single layer neural network classifier with softmax activation.

train_dataset consists of num_train many labeled examples, where the labels are in {0,1,...,num_classes-1}.
test_dataset consists of num_test many labeled examples, where the labels are in {0,1,...,num_classes-1}.
epochs the number of epochs.
num_classes the number of classes.
optimizer a tf.keras optimizer.
loss a tf.keras loss function.
batch_size a positive integer.
kernel_regularizer a regularization function.

List of test accuracies (one for each epoch) on test_dataset of model trained on train_dataset.