Text Classification
Transformers
Safetensors
yield-weather-soil
crop-yield
multi-temporal
regression
yield-estimation
custom_code
Instructions to use ICICLE-AI/yield-estimation with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use ICICLE-AI/yield-estimation with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="ICICLE-AI/yield-estimation", trust_remote_code=True)# Load model directly from transformers import AutoModelForSequenceClassification model = AutoModelForSequenceClassification.from_pretrained("ICICLE-AI/yield-estimation", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update config.json
Browse files- config.json +12 -18
config.json
CHANGED
|
@@ -268,32 +268,26 @@
|
|
| 268 |
],
|
| 269 |
"y_mean": 166.3075714111328,
|
| 270 |
"y_std": 30.291048049926758,
|
| 271 |
-
|
| 272 |
-
"auto_map": {
|
| 273 |
-
"AutoConfig": "configuration_yield.YieldConfig",
|
| 274 |
-
"AutoModel": "modeling_yield.YieldForRegression",
|
| 275 |
-
"AutoModelForSequenceClassification": "modeling_yield.YieldForSequenceClassification",
|
| 276 |
-
"AutoTokenizer": [
|
| 277 |
-
"tokenization_yield.YieldTokenizer",
|
| 278 |
-
null
|
| 279 |
-
],
|
| 280 |
-
"AutoProcessor": "processing_yield.YieldProcessor"
|
| 281 |
-
},
|
| 282 |
"problem_type": "regression",
|
| 283 |
"num_labels": 1,
|
|
|
|
| 284 |
"id2label": {
|
| 285 |
"0": "YIELD_BU_ACRE"
|
| 286 |
},
|
|
|
|
| 287 |
"label2id": {
|
| 288 |
"YIELD_BU_ACRE": 0
|
| 289 |
},
|
|
|
|
| 290 |
"tokenizer_class": "YieldTokenizer",
|
| 291 |
-
|
| 292 |
-
|
| 293 |
-
|
| 294 |
-
|
| 295 |
-
|
| 296 |
-
|
| 297 |
-
|
|
|
|
| 298 |
}
|
| 299 |
}
|
|
|
|
| 268 |
],
|
| 269 |
"y_mean": 166.3075714111328,
|
| 270 |
"y_std": 30.291048049926758,
|
| 271 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 272 |
"problem_type": "regression",
|
| 273 |
"num_labels": 1,
|
| 274 |
+
|
| 275 |
"id2label": {
|
| 276 |
"0": "YIELD_BU_ACRE"
|
| 277 |
},
|
| 278 |
+
|
| 279 |
"label2id": {
|
| 280 |
"YIELD_BU_ACRE": 0
|
| 281 |
},
|
| 282 |
+
|
| 283 |
"tokenizer_class": "YieldTokenizer",
|
| 284 |
+
|
| 285 |
+
"auto_map": {
|
| 286 |
+
"AutoConfig": "configuration_yield.YieldConfig",
|
| 287 |
+
"AutoModelForSequenceClassification": "modeling_yield.YieldForSequenceClassification",
|
| 288 |
+
"AutoTokenizer": [
|
| 289 |
+
"tokenization_yield.YieldTokenizer",
|
| 290 |
+
null
|
| 291 |
+
]
|
| 292 |
}
|
| 293 |
}
|