draw.py 文件源码

def __init__(self, dimX, dimReadAttent, dimWriteAttent, dimRNNEnc, dimRNNDec, dimZ, runSteps, inputData, valData=None, testData=None, pathToWeights=None):
        self.dimX = dimX
        self.dimReadAttent = dimReadAttent
        self.dimWriteAttent = dimWriteAttent
        self.dimRNNEnc = dimRNNEnc
        self.dimRNNDec = dimRNNDec
        self.dimZ = dimZ
        self.runSteps = runSteps
        self.pathToWeights = pathToWeights

        self.n_batches = inputData.shape[0] / batch_size
        self.train_data = theano.shared(inputData)
        del inputData

        if valData != None:
            self.n_val_batches = valData.shape[0] / batch_size
            self.val_data = theano.shared(valData)
            del valData

        if testData != None:
            self.n_test_batches = testData.shape[0] / batch_size
            self.test_data = theano.shared(testData)
            del testData

        self._kl_final, self._logpxz, self._log_likelihood, self._c_ts, self._c_ts_gener, self._x, self._run_steps, self._updates_train, self._updates_gener, self._read_attent_params, self._write_attent_params, self._write_attent_params_gener, self._params = build_lstm_attention_vae(self.dimX, self.dimReadAttent, self.dimWriteAttent, self.dimRNNEnc, self.dimRNNDec, self.dimZ, self.runSteps, self.pathToWeights)