import sys
from learning_functions.perform_learning import perform_learning
from keras_models.identification import identification_unet

# inputs to the model
model_params = {'kernel_size': (3, 3),
                'filters': 32,
                'learning_rate': 0.001}

perform_learning(training_sample_dir=sys.argv[1],
                 val_sample_dir=sys.argv[2],
                 batch_size=32,
                 three_d=False,
                 sample_channels=8,
                 categorise=False,
                 output_classes=1,
                 shuffle=True,
                 model_func=identification_unet,
                 model_params=model_params,
                 epochs=35,
                 model_path=sys.argv[3],
                 checkpoint_path="checkpoints/identification_model/identification_model.{epoch:02d}.h5",
                 log_name="identification")
