INNER CODE UNIT · Python
plot_prediction
aimclub/FEDOT · fedot/api/main.py:402
def plot_prediction(self, in_sample: Optional[bool] = None, target: Optional[Any] = None):
"""Plots prediction obtained from a graph.
Args:
in_sample: if current prediction is in_sample (for time-series forecasting), plots predictions as future
values.
target: user-specified name of target variable for :class:`MultiModalData`.
"""
task = self.params.task
if self.prediction is not None:
if task.task_type == TaskTypesEnum.ts_forecasting:
in_sample = in_sample or self._is_in_sample_prediction
plot_forecast(self.test_data, self.prediction, in_sample, target)
elif task.task_type == TaskTypesEnum.regression:
plot_biplot(self.prediction)
elif task.task_type == TaskTypesEnum.classification:
self.predict_proba(self.test_data)