agroaid-models / README.md
arinbalyan's picture
Upload folder using huggingface_hub
10da115 verified
|
Raw
History Blame Contribute Delete
724 Bytes
metadata
license: mit
library_name: pytorch

AgroAid Models

Trained models for the AgroAid smart farming application:

Crop Disease Classifier

  • Model: ResNet-18
  • Dataset: PlantVillage (54 classes, 87K images)
  • Training notebook: arinbalyan/agroaid-crop-disease-classification
  • Status: Training on Kaggle

Crop Yield Predictor

  • Model: LightGBM / Random Forest
  • Dataset: FAO Crop Yield (Kaggle)
  • Training notebook: arinbalyan/agroaid-crop-yield-prediction
  • Status: Training on Kaggle

Crop Recommendation

  • Model: XGBoost
  • Dataset: Kaggle Crop Recommendation
  • Training notebook: arinbalyan/agroaid-crop-recommendation
  • Status: Training on Kaggle

Models will be uploaded after Kaggle training completes.