class_weights: null
learning_rate: 0.001
num_classes: 5