[Task] Update multi-task learning models and tests to use OutputBlock
instead of PredictionsTasks
#914
Closed
2 tasks done
Description
Currently, the Multi-task Learning models (
MMOEBlock
,CGCBlock
,PLEBLock
introduced/refactored in #902 ) and their tests are usingPredictionsTasks
.The PR #772 (in progress by @marcromeyn ) introduces a new API for multi-task learning outputs (
OutputBlock
), which will require changes in the multi-task learning tests and models.Tasks
OutputBlock
instead ofPredictionTasks
.MMOEBlock
,CGCBlock
andPLEBlock
need to be aware of the task names. They currently receive an ParallelPredictionBlock, but could potentially receive the schema and infer from there the task names.This task is a pre-requisite for
The text was updated successfully, but these errors were encountered: