mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-28 07:57:44 +00:00
31 lines
915 B
Python
31 lines
915 B
Python
|
from rdagent.core.conf import BenchmarkSettings
|
||
|
|
from rdagent.core.utils import import_class
|
||
|
|
from rdagent.benchmark.eval_method import FactorImplementEval
|
||
|
|
from rdagent.benchmark.data_process import load_eval_data
|
||
|
|
|
||
|
|
# 1.read the settings
|
||
|
|
bs = BenchmarkSettings()
|
||
|
|
|
||
|
|
# 2.read and prepare the eval_data
|
||
|
|
test_cases = load_eval_data(bs.bench_version)
|
||
|
|
|
||
|
|
# 3.declare the method to be tested and pass the arguments.
|
||
|
|
# TODO: Whether it is necessary to define two Eval method classes for two data type?
|
||
|
|
method_cls = import_class(bs.bench_method_cls)
|
||
|
|
generate_method = method_cls(bs.bench_method_extra_kwargs)
|
||
|
|
|
||
|
|
# 4.declare the eval method and pass the arguments.
|
||
|
|
eval_method = FactorImplementEval(
|
||
|
|
method=generate_method,
|
||
|
|
test_cases=test_cases,
|
||
|
|
catch_eval_except=True,
|
||
|
|
test_round=bs.bench_test_round,
|
||
|
|
)
|
||
|
|
|
||
|
|
# 5.run the eval
|
||
|
|
eval_method.eval()
|
||
|
|
|
||
|
|
# 6.save the result
|
||
|
|
eval_method.save(output_path = bs.bench_result_path)
|
||
|
|
|