|
--- |
|
metrics: |
|
- accuracy |
|
language: |
|
- en |
|
license: other |
|
pipeline_tag: image-classification |
|
--- |
|
# CNN Model for Kidney Disease Classification |
|
|
|
This is a Convolutional Neural Network (CNN) model for classifying kidney disease using medical images. |
|
|
|
## Model Architecture |
|
- **Base architecture**: ResNet / AlexNet |
|
- **Training data**: [Your Dataset Name] |
|
- **Evaluation metric**: Accuracy, F1-Score |
|
|
|
## Usage |
|
```python |
|
from transformers import AutoModel |
|
model = AutoModel.from_pretrained("yashpat85/ResNet01") |
