File size: 457 Bytes
4d6e8c2 fe4a4cb 4d6e8c2 |
1 2 3 4 5 6 7 8 9 |
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 AudioEvaluationRequest(BaseEvaluationRequest):
dataset_name: str = Field("rfcx/frugalai",
description="The name of the dataset on HuggingFace Hub") |