Spaces:
Sleeping
Sleeping
File size: 892 Bytes
4d6e8c2 fe4a4cb 4d6e8c2 fe4a4cb 4d6e8c2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from typing import Optional
from pydantic import BaseModel, Field
class BaseEvaluationRequest(BaseModel):
test_size: float = Field(0.2, ge=0.0, le=1.0, description="Size of the test split (between 0 and 1)")
test_seed: int = Field(42, ge=0, description="Random seed for reproducibility")
class TextEvaluationRequest(BaseEvaluationRequest):
dataset_name: str = Field("QuotaClimat/frugalaichallenge-text-train",
description="The name of the dataset on HuggingFace Hub")
class ImageEvaluationRequest(BaseEvaluationRequest):
dataset_name: str = Field("pyronear/pyro-sdis",
description="The name of the dataset on HuggingFace Hub")
class AudioEvaluationRequest(BaseEvaluationRequest):
dataset_name: str = Field("rfcx/frugalai",
description="The name of the dataset on HuggingFace Hub") |