corpus is a Dict

This commit is contained in:
svlandeg 2020-09-15 22:07:16 +02:00
parent 51fa929f47
commit 7336657662
1 changed files with 1 additions and 1 deletions

View File

@ -198,7 +198,7 @@ class ModelMetaSchema(BaseModel):
class ConfigSchemaTraining(BaseModel):
# fmt: off
vectors: Optional[StrictStr] = Field(..., title="Path to vectors")
corpus: Reader = Field(..., title="Reader for the training and dev data")
corpus: Dict[str, Reader] = Field(..., title="Reader for the training and dev data")
batcher: Batcher = Field(..., title="Batcher for the training data")
dropout: StrictFloat = Field(..., title="Dropout rate")
patience: StrictInt = Field(..., title="How many steps to continue without improvement in evaluation score")