twodlearn.templates.bayesnet module

class twodlearn.templates.bayesnet.EVGPEstimator(options=None, logger_path='tmp', session=None, **kargs)[source]

Bases: twodlearn.templates.bayesnet.VGPEstimator

model[source]

Autoinit with arguments [‘dataset’, ‘prior’]

train[source]

Autoinit with arguments [‘dataset’, ‘batch_size’]

class twodlearn.templates.bayesnet.ExplicitGpEstimator(options=None, logger_path='tmp', session=None, **kargs)[source]

Bases: twodlearn.templates.bayesnet.GpEstimator

Explicit gp estimator

fit(train_x, train_y, max_iter=100)[source]

fit the gaussian process to the training data. By default, the parameters of the kernel are optimized.

Parameters
  • train_x – input training data [batch_size, x_size].

  • train_y – output training data [batch_size, y_size].

  • max_iter – number of maximum iterations to run the optimization. Defaults to 100.

model[source]

Autoinit with arguments [‘train_x’, ‘train_y’]

test[source]
class twodlearn.templates.bayesnet.GpEstimator(options=None, logger_path='tmp', session=None, **kargs)[source]

Bases: twodlearn.templates.supervised.SupervisedEstimator

fit(train_x, train_y, max_iter=100)[source]

fit the gaussian process to the training data. By default, the parameters of the kernel are optimized.

Parameters
  • train_x – input training data [batch_size, x_size].

  • train_y – output training data [batch_size, y_size].

  • max_iter – number of maximum iterations to run the optimization. Defaults to 100.

get_save_data()[source]
model[source]

Autoinit with arguments [‘train_x’, ‘train_y’]

normalizer[source]
optimizer[source]

Autoinit with arguments [‘learning_rate’, ‘valid_freq’]

predict(x)[source]
test[source]
train[source]
valid[source]
class twodlearn.templates.bayesnet.VGPEstimator(options=None, logger_path='tmp', session=None, **kargs)[source]

Bases: twodlearn.templates.supervised.SupervisedEstimator

fit(dataset, max_iter=100, feed_dict=None)[source]
m[source]
model[source]

Autoinit with arguments [‘dataset’]

normalizer[source]
optimizer[source]

Autoinit with arguments [‘learning_rate’, ‘valid_freq’]

predict(x)[source]
test[source]

Autoinit with arguments [‘dataset’, ‘tolerance’]

train[source]

Autoinit with arguments [‘dataset’, ‘batch_size’]

valid[source]