#!/usr/bin/env python3
__all__ = [
    "Exact"
]
from .. import RunExactGP
from .._const import regs
[docs]class Exact(RunExactGP):
    """ExactGPModelの実行クラス
    Parameters
    ----------
    kernel : str or :obj:`gpytorch.kernels`, default :obj:`'RBFKernel'`
        使用するカーネル関数を指定する。下記から選択する。
        - :obj:`'CosineKernel'`
        - :obj:`'LinearKernel'`
        - :obj:`'MaternKernel'`
        - :obj:`'PeriodicKernel'`
        - :obj:`'RBFKernel'`
        - :obj:`'RQKernel'`
        - :obj:`'SpectralMixtureKernel'`
        基本はstrで指定されることを想定しているものの、 :obj:`gpytorch.kernels`
        を用いた自作のカーネル関数を入力することも可能
    likelihood : str, default :obj:`'GaussianLikelihood'`
        likelihoodとして使用するクラス名が指定される。
        - :obj:`'GaussianLikelihood'`, :obj:`'GL'` : likelihoodにガウス分布を仮定したガウス過程を行う場合
    optimizer : str, default :obj:`'Adam'`
        optimizerとして使用するクラス名が指定される。下記から選択する。
        - :obj:`'Adam'`
        - :obj:`'sgd'`
        - :obj:`'RMSprop'`
        - :obj:`'Adadelta'`
        - :obj:`'Adagrad'`
    mll : str, default :obj:`'ExactMarginalLogLikelihood'`
        確率分布の周辺化の方法のクラス名が指定される。
        - :obj:`'ExactMarginalLogLikelihood'`
    ard_option : bool, default True
        ARDカーネルを利用するかが指定される
        もし :obj:`kernel_coeff` を利用する場合 `ard_option=True` を選択する
    ker_conf : dict, default dict()
        カーネル関数に渡す設定
    mll_conf : dict, default dict()
        mllに渡す設定
    opt_conf : dict, default dict()
        optimizerに渡す設定
    random_state : int, default None
        seedの固定
    """
    def __init__(self, *args, **kwargs):
        if kwargs.get('likelihood') in regs:
            raise ValueError(f"Classifierに{kwargs.get('likelihood')}は使えない")
        else:
            raise ValueError("Exactでは分類問題を解けない")
            # super(Exact, self).__init__(*args, **kwargs)
[docs]    def set_model(self, *args, **kwargs):
        super(Exact, self).set_model(*args, **kwargs) 
[docs]    def fit(self, *args, **kwargs):
        super(Exact, self).fit(*args, **kwargs) 
[docs]    def predict(self, *args, **kwargs):
        """予測用メソッド
        Parameters
        ----------
        X : np.array or torch.tensor
            入力説明変数
        cl : float default 0.6827(1sigma)
            信頼区間
        sample_num : int default None
            yのサンプル数
        sample_f_num : int default None
            fのサンプル数
        Returns
        -------
        output : object
            予測された目的変数のオブジェクト。下記の属性が用意されている。
            - output.samples : 入力説明変数に対する予測値yのサンプル(sample_num個サンプルされる)
            - output.samples_f : 入力説明変数に対する予測関数fのサンプル(sample_f_num個サンプルされる)
            - output.probs : 2値分類の予測確率。
        """
        return super(Exact, self).predict(*args, **kwargs) 
[docs]    def save(self, *args, **kwargs):
        super(Exact, self).save(*args, **kwargs) 
[docs]    def load(self, *args, **kwargs):
        super(Exact, self).load(*args, **kwargs) 
[docs]    def kernel_coeff(self, *args, **kwargs):
        super(Exact, self).kernel_coeff(*args, **kwargs) 
[docs]    def plot_kernel(self, *args, **kwargs):
        super(Exact, self).plot_kernel(*args, **kwargs)