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)# pip install -U transformers accelerate # 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 modeling_yield.py
Browse files- modeling_yield.py +2 -2
modeling_yield.py
CHANGED
|
@@ -10,8 +10,8 @@ from transformers import PreTrainedModel
|
|
| 10 |
from transformers.modeling_outputs import ModelOutput
|
| 11 |
from dataclasses import dataclass
|
| 12 |
|
| 13 |
-
from configuration_yield import YieldConfig
|
| 14 |
-
from yield_transformer import YieldTransformer
|
| 15 |
|
| 16 |
|
| 17 |
@dataclass
|
|
|
|
| 10 |
from transformers.modeling_outputs import ModelOutput
|
| 11 |
from dataclasses import dataclass
|
| 12 |
|
| 13 |
+
from .configuration_yield import YieldConfig
|
| 14 |
+
from .yield_transformer import YieldTransformer
|
| 15 |
|
| 16 |
|
| 17 |
@dataclass
|