dq.set_tasks_for_run

A multi-label text classification function

def set_tasks_for_run(tasks: List[str], binary: bool = False) -> None:
Sets the task names for the run (multi-label case only). Sets the order of the tasks with respect to the model's probability vectors.
Arguments
Text
tasks
An ordered list of tasks ie ['happiness','anger','sadness']. This order MUST match the order of probability vectors for each task that the model outputs.
This also must match the order of the labels logged in set_labels_for_run (meaning that the first list of labels must be the labels of the first task passed in here)
binary
Whether this is a binary multi label run. If True, tasks will also be set as your labels, and you should NOT call dq.set_labels_for_run it will be handled for you. Default False

Arguments	Text
`tasks`	An ordered list of tasks ie `['happiness','anger','sadness']`. This order MUST match the order of probability vectors for each task that the model outputs. This also must match the order of the labels logged in set_labels_for_run (meaning that the first list of labels must be the labels of the first task passed in here)
`binary`	Whether this is a binary multi label run. If `True`, tasks will also be set as your labels, and you should NOT call `dq.set_labels_for_run` it will be handled for you. Default `False`

dq.set_labels_for_run

dq.log_model_outputs

Last modified 9mo ago