File size: 798 Bytes
21df233
 
256861b
 
 
ba92d02
f245958
 
21df233
 
931bca5
21df233
 
 
5ee97df
21df233
 
 
931bca5
21df233
 
931bca5
21df233
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
---
library_name: keras
tags:
- image-classification
- keras

---

## Model description

It's pretty simple, the model tries to differentiate between dogs/cats. 

## Intended uses & limitations

Only JPGs can be used and they are auto resized. **Meant for dogs and cats only.**

## Training and evaluation data

Combined training data from multiple sources. Downloaded around 3000 cat and dog photos from r/dogs and r/cats. Curated kaggle datasets into one large dataset. 

## Training procedure
Trained using an early stopping for overfitting

### Training hyperparameters

The following hyperparameters were used during training:
- optimizer: {'name': 'Adam', 'learning_rate': 0.001, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
- training_precision: float32