merlin.models.tf.PredictionTasks

merlin.models.tf.PredictionTasks(schema: merlin.schema.schema.Schema, task_blocks: Optional[Union[keras.engine.base_layer.Layer, Dict[str, keras.engine.base_layer.Layer]]] = None, task_weight_dict: Optional[Dict[str, float]] = None, bias_block: Optional[keras.engine.base_layer.Layer] = None, loss_reduction=<function reduce_mean>, **kwargs) → merlin.models.tf.prediction_tasks.base.ParallelPredictionBlock