Nidan / server /tasks /task2_multiclass.py
Saivats's picture
feat: enhance environment with curriculum rewards and multi-metric graders
3ba0ffc
Raw
History Blame Contribute Delete
466 Bytes
from __future__ import annotations
from server.tasks.task_base import TaskConfig
TASK2_CONFIG = TaskConfig(
task_id="task2",
name="Multi-class Chest Conditions",
classes=["normal", "pneumonia", "covid", "tuberculosis"],
pool_size=400,
budget=30,
success_threshold=0.65,
rare_classes=["covid", "tuberculosis"],
modality="xray",
body_part="chest",
variable_cost=True,
base_annotation_cost=1,
rare_annotation_cost=2,
)