def create_model()

in source_directory/training/training_script.py [0:0]

• 14 lines of code
• 2 McCabe index (conditional complexity)

def create_model():
    
    # create model with resnet base
    resnet_base = ResNet50V2(include_top=False, weights="imagenet", input_shape=(32,32,3), pooling='avg',)
    for layer in resnet_base.layers:
        layer.trainable = True

    model_input = Input(shape=(32,32,3), name='image_input')
    model = resnet_base(model_input)
    model = Dense(256)(model)
    model = BatchNormalization()(model)
    model = Activation('relu')(model)
    model = Dropout(rate=0.3)(model)
    model = Dense(10)(model)
    outputs = Activation('softmax')(model)

    model = Model(inputs=model_input, outputs=outputs)
    
    return model