README.md

metadata

tags:
  - image-classification
  - pytorch
  - huggingpics
metrics:
  - accuracy
model-index:
  - name: pneumonia-bielefeld-dl-course
    results:
      - task:
          name: Image Classification
          type: image-classification
        metrics:
          - name: Accuracy
            type: accuracy
            value: 0.8456632494926453

pneumonia-bielefeld-dl-course

This registry contains the model for making pneumonia predictions and was prepared for Bielefeld University Deep Learning course homework.

The code used for this implementation mostly comes from here: https://github.com/nateraw/huggingpics it was a ready pipeline for model fine-tuning with huggingface and PyTorch Lightning for another dataset.