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, task_pre_dict: Optional[Dict[str, merlin.models.tf.core.base.Block]] = None, bias_block: Optional[keras.engine.base_layer.Layer] = None, **kwargs) → merlin.models.tf.prediction_tasks.base.ParallelPredictionBlock[source]

Creates Multi-task prediction Blocks from schema

Parameters

  • schema (Schema) – The Schema with the input features

  • task_blocks (Optional[Union[Layer, Dict[str, Layer]]], optional) – Task blocks to be used for prediction, by default None

  • task_weight_dict (Optional[Dict[str, float]], optional) – Dict where keys are target feature names and values are weights for each task, by default None

  • task_pre_dict (Optional[Dict[str, Block]], optional) – Dict where keys are target feature names and values are Blocks to be used as pre for those tasks

  • bias_block (Optional[Layer], optional) – Bias block to be used for prediction, by default None