Python LubyBenchmark.read_instance_set Examples

Example #1

File: test_luby_benchmark.py Project: rishan92/DACBench

 def test_read_instances(self):
     bench = LubyBenchmark()
     bench.read_instance_set()
     self.assertTrue(len(bench.config.instance_set) == 1)
     self.assertTrue(len(bench.config.instance_set[0]) == 2)
     self.assertTrue(bench.config.instance_set[0] == [0, 0])
     bench2 = LubyBenchmark()
     env = bench2.get_environment()
     self.assertTrue(len(env.instance_set[0]) == 2)
     self.assertTrue(env.instance_set[0] == [0, 0])
     self.assertTrue(len(env.instance_set) == 1)

Example #2

File: test_instance_sampling_wrapper.py Project: rishan92/DACBench

    def test_fit(self):
        bench = LubyBenchmark()
        bench.config.instance_set_path = "../instance_sets/luby/luby_train.csv"
        bench.read_instance_set()
        instances = bench.config.instance_set
        env = bench.get_environment()

        wrapped = InstanceSamplingWrapper(env, instances=instances)
        samples = []
        for _ in range(100):
            samples.append(wrapped.sampling_function())
        mi1 = mutual_info_score(np.array(instances)[:, 0], np.array(samples)[:, 0])
        mi2 = mutual_info_score(np.array(instances)[:, 1], np.array(samples)[:, 1])

        self.assertTrue(mi1 > 0.99)
        self.assertTrue(mi1 != 1)
        self.assertTrue(mi2 > 0.99)
        self.assertTrue(mi2 != 1)