ModelTaskParams
ModelTaskParams
Examples
binary_params = ModelTaskParams(
binary_classification_threshold=0.5,
target_class_order=[“negative”, “positive”]
)multiclass_params = ModelTaskParams(
target_class_order=[“class_a”, “class_b”, “class_c”],
class_weights=[0.3, 0.5, 0.2],
weighted_ref_histograms=True
)ranking_params = ModelTaskParams(
group_by=”query_id”,
top_k=10,
target_class_order=[“not_relevant”, “relevant”, “highly_relevant”]
)Last updated
Was this helpful?