Store feature and class names (if provided) in InputData and transfer them to OutputData (predictions)
Created by: MorrisNein
When InputData
is retrieved from a source with explicit feature and target class names (e.g. .csv
file or Pandas DataFrame
), it should remember their labels. The labels should also be included in OutputData
(predictions) of the pipeline.
One way to do it is to use SupplementaryData
, which is already included as an attribute of InputData
and OutputData
. Also, consider transferring categorial feature labels the same way.