https://github.com/GPflow/GPflow
Raw File
Tip revision: ea67c67014dd1548564ab03e7a0fe99591e11e67 authored by John Bradshaw on 04 October 2017, 08:40:37 UTC
Random features -- improved demo to also show Thompson sampling.
Tip revision: ea67c67
gpflow_testcase.py
import unittest
import tensorflow as tf

class GPflowTestCase(tf.test.TestCase):
    """
    Wrapper for TestCase to avoid massive duplication of resetting
    Tensorflow Graph.
    """

    _multiprocess_can_split_ = True

    def tearDown(self):
        tf.reset_default_graph()
        super(GPflowTestCase, self).tearDown()
back to top