Python BenchmarkResultの例

コード例 #1

ファイル: test_benchmark_result.py プロジェクト: EmuKit/emukit

def test_benchmark_result_initialization():
    br = BenchmarkResult(["loop1", "loop2"], 5, ["mae", "mse"])

    for loop in ["loop1", "loop2"]:
        for metric in ["mae", "mse"]:
            metric_data = br.extract_metric_as_array(loop, metric)

            assert metric_data.size == 0
            assert metric_data.shape == (5, 0)

コード例 #2

ファイル: test_benchmark_result.py プロジェクト: zhucer2003/emukit

def test_benchmark_result_initialization():
    br = BenchmarkResult(['loop1', 'loop2'], 5, ['mae', 'mse'])

    for loop in ['loop1', 'loop2']:
        for metric in ['mae', 'mse']:
            metric_data = br.extract_metric_as_array(loop, metric)

            assert metric_data.size == 0
            assert metric_data.shape == (5, 0)

コード例 #3

ファイル: test_benchmark_result.py プロジェクト: zhucer2003/emukit

def test_benchmark_result_extract_invalid_names():
    br = BenchmarkResult(['loop1', 'loop2'], 5, ['mae', 'mse'])

    with pytest.raises(KeyError):
        br.extract_metric_as_array('invalid loop', 'mse')

    with pytest.raises(KeyError):
        br.extract_metric_as_array('loop1', 'invalid metric')

コード例 #4

ファイル: test_benchmark_result.py プロジェクト: EmuKit/emukit

def test_benchmark_result_extract_invalid_names():
    br = BenchmarkResult(["loop1", "loop2"], 5, ["mae", "mse"])

    with pytest.raises(KeyError):
        br.extract_metric_as_array("invalid loop", "mse")

    with pytest.raises(KeyError):
        br.extract_metric_as_array("loop1", "invalid metric")

コード例 #5

ファイル: test_benchmark_plot.py プロジェクト: EmuKit/emukit

def benchmark_result():
    br = BenchmarkResult(["loop1", "loop2"], 2, ["mae", "mse"])

    br.add_results("loop1", 0, "mae", np.array([1, 2, 3]))
    br.add_results("loop1", 0, "mse", np.array([1, 2]))
    br.add_results("loop1", 1, "mae", np.array([4, 5, 6]))
    br.add_results("loop1", 1, "mse", np.array([1, 2]))

    br.add_results("loop2", 0, "mae", np.array([1]))
    br.add_results("loop2", 0, "mse", np.array([2]))
    br.add_results("loop2", 1, "mae", np.array([4]))
    br.add_results("loop2", 1, "mse", np.array([1]))

    return br

コード例 #6

ファイル: test_benchmark_result.py プロジェクト: zhucer2003/emukit

def test_benchmark_result_add_data():
    br = BenchmarkResult(['loop1'], 2, ['mae', 'mse'])

    br.add_results('loop1', 0, 'mae', np.array([1, 2, 3]))
    br.add_results('loop1', 0, 'mse', np.array([1, 2]))
    br.add_results('loop1', 1, 'mae', np.array([4, 5, 6]))
    br.add_results('loop1', 1, 'mse', np.array([1, 2]))

    mae_data = br.extract_metric_as_array('loop1', 'mae')

    assert_array_equal(mae_data, np.array([[1, 2, 3], [4, 5, 6]]))

    mse_data = br.extract_metric_as_array('loop1', 'mse')

    assert_array_equal(mse_data, np.array([[1, 2], [1, 2]]))

コード例 #7

ファイル: test_benchmark_result.py プロジェクト: EmuKit/emukit

def test_benchmark_result_add_data():
    br = BenchmarkResult(["loop1"], 2, ["mae", "mse"])

    br.add_results("loop1", 0, "mae", np.array([1, 2, 3]))
    br.add_results("loop1", 0, "mse", np.array([1, 2]))
    br.add_results("loop1", 1, "mae", np.array([4, 5, 6]))
    br.add_results("loop1", 1, "mse", np.array([1, 2]))

    mae_data = br.extract_metric_as_array("loop1", "mae")

    assert_array_equal(mae_data, np.array([[1, 2, 3], [4, 5, 6]]))

    mse_data = br.extract_metric_as_array("loop1", "mse")

    assert_array_equal(mse_data, np.array([[1, 2], [1, 2]]))

コード例 #8

ファイル: test_benchmark_plot.py プロジェクト: zhucer2003/emukit

def benchmark_result():
    br = BenchmarkResult(['loop1', 'loop2'], 2, ['mae', 'mse'])

    br.add_results('loop1', 0, 'mae', np.array([1, 2, 3]))
    br.add_results('loop1', 0, 'mse', np.array([1, 2]))
    br.add_results('loop1', 1, 'mae', np.array([4, 5, 6]))
    br.add_results('loop1', 1, 'mse', np.array([1, 2]))

    br.add_results('loop2', 0, 'mae', np.array([1]))
    br.add_results('loop2', 0, 'mse', np.array([2]))
    br.add_results('loop2', 1, 'mae', np.array([4]))
    br.add_results('loop2', 1, 'mse', np.array([1]))

    return br