Source code for immuneML.workflows.steps.MLMethodAssessmentParams

from pathlib import Path

from immuneML.data_model.dataset.Dataset import Dataset
from immuneML.environment.Label import Label
from immuneML.ml_methods.MLMethod import MLMethod
from immuneML.ml_metrics.Metric import Metric
from immuneML.workflows.steps.StepParams import StepParams


[docs]class MLMethodAssessmentParams(StepParams): def __init__(self, method: MLMethod, dataset: Dataset, metrics: set, optimization_metric: Metric, label: Label, path: Path, split_index: int, predictions_path: Path, ml_score_path: Path): self.method = method self.dataset = dataset self.metrics = metrics self.optimization_metric = optimization_metric self.path = path self.label = label self.split_index = split_index self.predictions_path = predictions_path self.ml_score_path = ml_score_path