Update README.md

README.md

@@ -74,35 +74,28 @@ All models were trained in bf16 mixed precision and were tuned for 24 hours on a
 
 ## Evaluation
 
-<!-- This section describes the evaluation protocols and provides the results. -->
-
-### Testing Data, Factors & Metrics
-
-#### Testing Data
+### Testing Data
 
 <!-- This should link to a Dataset Card if possible. -->
 
 Refer to the Dataset card to find the relevant testing data, as different kinds exist.
 
-#### Metrics
-
-<!-- These are the evaluation metrics being used, ideally with a description of why. -->
-
-[More Information Needed]
-
-### Results
-
-The ResNet-50 and ResNet-152 architectures were found to work the best for our task, as they were the only two that were able to generalize between the augmented training data and the un-augmented validation data.
 
 #### Summary
 
 The `base` directory contains all trained models with base hyperparameters. The `tuned` directory contains the models after hyperparameter tuning. The `hyperparameters` folder itself contains each model's tuned hyperparameters in a JSON file, and the `inference_examples` directory contains examples for how to inference our models in safetensors format.
 
-## Model Examination [optional]
+## Model Examination
 
-<!-- Relevant interpretability work for the model goes here -->
+For all models the Captum library was used to aid in explainability of all models. For the MLP models, the following explainability algorithms were used:
 
-[More Information Needed]
+- GradientSHAP
+- DeepLIFT
+- Feature ablation
+
+For the CNN architectures, the GradCAM algorithm was used to create a heatmap of areas the model used to create its permeability estimate.
+
+For more details and results, refer to the paper.
 
 ## Environmental Impact
 
@@ -144,16 +137,6 @@ All models were written in PyTorch Lightning.
 
 [More Information Needed]
 
-## Glossary [optional]
-
-<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
-
-[More Information Needed]
-
-## More Information [optional]
-
-[More Information Needed]
-
 ## Model Card Authors
 
 Josh Gregory