--- a +++ b/Classifier/Model/config.json @@ -0,0 +1,76 @@ +{ + "classifier": { + "api": { + "server": "", + "port": 1234 + }, + "core": [ + "Train", + "Server", + "Client", + "Classify" + ], + "data": { + "dim": 100, + "file_type": ".jpg", + "labels": [0, 1], + "rotations": 10, + "seed": 2, + "split": 0.3, + "valid": "C:/Peter Moss AML Leukemia Research/Dataset/valid", + "test_data": [ + "im006_1.jpg", + "im020_1.jpg", + "im024_1.jpg", + "im026_1.jpg", + "im028_1.jpg", + "im031_1.jpg", + "im035_0.jpg", + "im041_0.jpg", + "im047_0.jpg", + "im053_1.jpg", + "im057_1.jpg", + "im060_1.jpg", + "im063_1.jpg", + "im069_0.jpg", + "im074_0.jpg", + "im088_0.jpg", + "im095_0.jpg", + "im099_0.jpg", + "im101_0.jpg", + "im106_0.jpg" + ], + "train": "C:/Peter Moss AML Leukemia Research/Dataset/train", + "valid_types": [ + ".JPG", + ".JPEG", + ".PNG", + ".GIF", + ".jpg", + ".jpeg", + ".png", + ".gif" + ] + }, + "model_params": { + "hidden_units": 500, + "input_features": 2048, + "output_units": 2, + "gamma": 0.1, + "weights": "Model/leukemia_model.pth", + "plot_loss": "Model/Plots/loss.png", + "plot_auc": "Model/Plots/auc.png", + "plot_loss_and_acc": "Model/Plots/loss_acc.png", + "confusion_matrix": "Model/Plots/confusion_matrix.png" + }, + "cnn_model": { + + }, + "train": { + "batch": 10, + "decay_adam": 1e-5, + "epochs": 100, + "learning_rate_adam": 0.0001 + } + } +} \ No newline at end of file