Source code for shamo.hd_tdcs.simulation.parametric.surrogate

from shamo.core.surrogate import SurrMaskedScalarNii


[docs]class SurrMaskedScalarNiiJ(SurrMaskedScalarNii):
[docs]    @classmethod
    def fit(cls, name, parent_path, sol, **kwargs):
        return super().fit(name, parent_path, sol, suffix="j", **kwargs)


[docs]class SurrMaskedScalarNiiMagJ(SurrMaskedScalarNii):
[docs]    @classmethod
    def fit(cls, name, parent_path, sol, **kwargs):
        return super().fit(name, parent_path, sol, suffix="mag_j", **kwargs)


[docs]class SurrMaskedScalarNiiV(SurrMaskedScalarNii):
[docs]    @classmethod
    def fit(cls, name, parent_path, sol, **kwargs):
        return super().fit(name, parent_path, sol, suffix="v", **kwargs)