New! Use Simple ML for Sheets to apply machine learning to the data in your Google Sheets Read More

tfdf.keras.MultiTaskItem

A single task in a multi-task configuration.

tfdf.keras.MultiTaskItem(
    label, task=1, primary=True, output=None
)

Models trained to predict label with primary=False are used to help the training of models with primary=True. If primary is true for all the tasks, each task will be trained independently.

label Key of the label.
task Task for the label.
primary The predictions of primary tasks are returned by "model.predict()", the output of non-primary tasks are not returned by "model.predict()". Non-primary tasks can be used to improve the primary task quality.
output If None, "model.predict" returns a dictionary of prediction indexed by "label" values. If set, "model.predict" returns a dictionary of predictions indexed by "output".

label A namedtuple alias for field number 0
task A namedtuple alias for field number 1
primary A namedtuple alias for field number 2
output A namedtuple alias for field number 3
prediction_key