---
license: mit
pipeline_tag: image-classification
datasets:
- rokmr/pets
---

My first CoreML ImageClassifier model using the [pets dataset](https://huggingface.co/datasets/rokmr/pets)

- **[Model Prediction]**
  - **Input**:
    - A Color Image (360 * 360)
  - **Output**:
    - Target: String
      - eg: "Cat"
    - Target Probability: Dictionary<String, Double>
      - ```json
        {
          "Cat" : 0.88,
          "Rabbit" : 0.0,
          "Dog" : 0.08
        }
        ```

- **[Made Using]**
  - Apple's [**CreateML** ![image/png](https://developer.apple.com/assets/elements/icons/create-ml/create-ml-96x96_2x.png)](https://developer.apple.com/documentation/createml/mlimageclassifier)

- **[Requirements]**
  - macOS 14.0+
  - iOS 17.0+
  - tvOS 17.0+

- **[Sample Usage]**:
  - [**GitHub**](https://github.com/c2p-cmd/PetsClassifierSwift/)

- **[Prediction Results]**
![image/png](https://cdn-uploads.huggingface.co/production/uploads/656968f81d7c2ca7b7b112c8/cvRhux2-fPQ0Bp_SUpK7d.png)