Source code for classification_report.return_types

from typing import List, Union

import numpy as np
import torch

[docs]LossType = Union[float, np.ndarray, torch.Tensor]
ActualType = PredictionType = Union[np.ndarray, torch.Tensor]
[docs]TrainType = List[str]