Upload config.json

config.json

@@ -0,0 +1,39 @@
+{
+  "model_name": "ResNet50",
+  "input_shape": [180, 180, 3],
+  "num_classes": 5,
+  "layers": [
+    {
+      "type": "ResNet50",
+      "include_top": false,
+      "weights": "imagenet"
+    },
+    {
+      "type": "Flatten"
+    },
+    {
+      "type": "Dense",
+      "units": 512,
+      "activation": "relu"
+    },
+    {
+      "type": "Dense",
+      "units": 5,
+      "activation": "softmax"
+    }
+  ],
+  "training": {
+    "batch_size": 8,
+    "epochs": 10,
+    "optimizer": "Adam",
+    "learning_rate": 0.001,
+    "loss": "categorical_crossentropy",
+    "metrics": ["accuracy"]
+  },
+  "dataset": {
+    "url": "https://storage.googleapis.com/download.tensorflow.org/example_images/flower_photos.tgz",
+    "validation_split": 0.25,
+    "image_size": [180, 180],
+    "label_mode": "categorical"
+  }
+}