Create top layer

create_top_classifier_layer.py

malaria_model = tf.keras.Sequential([features_extractor_layer,
                                     layers.Dense(training_generator.num_classes, 
                                                  activation='softmax')
                                    ])
malaria_model.summary()

malaria_model.compile(optimizer=tf.keras.optimizers.RMSprop(lr=0.001), 
                      loss='categorical_crossentropy',
                      metrics=['acc'])
"""
_________________________________________________________________
Layer (type)                 Output Shape              Param #   
=================================================================
lambda_1 (Lambda)            (None, 1792)              0         
_________________________________________________________________
dense (Dense)                (None, 2)                 3586      
=================================================================
Total params: 3,586
Trainable params: 3,586
Non-trainable params: 0
_________________________________________________________________
"""