Upload 108 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+brain-tumor-train.csv filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -0,0 +1,10 @@
+# Use the official TensorFlow Docker image as the base image
+FROM tensorflow/tensorflow:latest
+
+WORKDIR /
+
+# Copy all files from the current directory to the Docker image
+COPY . .
+
+# Install any additional dependencies (you can add more if needed)
+RUN pip install --no-cache-dir Pillow argparse pandas numpy scikit-learn

README

@@ -0,0 +1,11 @@
+# Train
+python train.py --dataset_path brain-tumor-train.csv --model_save_path model.h5
+
+# Generate GRadients
+python gen_gradients.py  --image_dir  /brain_tumor_classifier_dataset --gradients_save_path /gradients --model_path model.h5
+
+# Update Model
+python update_model.py --model_path model.h5 --saved_gradients gradients --dataset_path brain-tumor-train.csv
+
+# Inference
+python inference.py --model_path model.h5 --image_dir brain_tumor_classifier_dataset --output_json outputs.json

brain-tumor-train.csv

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d41020860e13cda2c900e4f7c2294b3b4f0e54295fcf621b584318ea465aca41
+size 199028317

brain_tumor_classifier.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8c18c9f8e6229690995ba23905be9f8ae1eda625d81f31fee92ddca28c078885
+size 25564344