Source code for qudas.pipeline.steps.iterator_mixin

from abc import ABC, abstractmethod
from typing import Any, Tuple
from .base import BaseStep


[docs]class IteratorMixin(BaseStep, ABC): """ パイプラインステップ内で使用するイテレータ用のMixinクラス。 このMixinクラスは次のパラメータセットを生成する `next_params` メソッドの実装を要求します。 """ def __init__(self, loop_num: int = 1) -> None: super().__init__() self.loop_num = loop_num # デフォルトで1回のループ self.models = None self.results = None
[docs] @abstractmethod def next_params(self, X: Any, y: Any = None, **iter_params) -> Tuple[Any, Any]: """ 次のパラメータセットを生成する抽象メソッド。 Args: X (Any): 入力データ。 y (Any): ターゲットデータ。 **iter_params: 追加のパラメータ。 Returns: Tuple[Any, Any]: 次の入力データとターゲットデータ。 """ pass