Python set_default_positive_bijector Examples

Example #1

def test_positive_bijector_error(value, error_msg):
    with pytest.raises(ValueError, match=error_msg):
        set_default_positive_bijector(value)

Example #2

def test_positive_bijector_setting(value):
    set_default_positive_bijector(value)
    assert default_positive_bijector() == value.lower()

Example #3

File: vaele_config.py Project: constantino-garcia/vaele

import os

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = ""

import gpflow
from gpflow.config import default_float
from gpflow.config import set_default_positive_bijector
import numpy as np
import tensorflow as tf
import tensorflow.keras.backend as backend

set_default_positive_bijector('softplus')

_VAELE_DEFAULT_FLOAT_ = 'float64'
_VAELE_DEFAULT_JITTER_ = 1e-6  # 1e-6 is gpflow default jitter


def _translate_str_float_to_np(float_str):
    _STR_FLOAT_TO_NP_ = {
        'float16': np.float16,
        'float32': np.float32,
        'float64': np.float64
    }
    if float_str not in _STR_FLOAT_TO_NP_:
        raise ValueError('Invalid float string')
    return _STR_FLOAT_TO_NP_[float_str]


backend.set_floatx(_VAELE_DEFAULT_FLOAT_)
gpflow.config.set_default_float(