コード例 #1
0
ファイル: states.py プロジェクト: mackelab/pyLDS_dev
    def __init__(self,
                 model,
                 T=None,
                 data=None,
                 stateseq=None,
                 generate=True,
                 initialize_from_prior=False,
                 initialize_to_noise=True):
        self.model = model
        self.data = data

        self.T = T if T else data.shape[0]
        self.data = data
        self._obs_scheme = ObservationScheme(p=self.p, T=self.T)

        self._normalizer = None

        if stateseq is not None:
            self.stateseq = stateseq
        elif generate:
            if data is not None and not (initialize_from_prior
                                         or initialize_to_noise):
                self.resample()
            else:
                if initialize_from_prior:
                    self.generate_states()
                else:
                    self.stateseq = np.random.normal(size=(self.T, self.n))