evaluator.py 1.6 KB
Newer Older
Y
Yu Yang 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
# Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserved
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import paddle.trainer_config_helpers.evaluators as evs
import inspect
from config_base import __convert_to_v2__

__all__ = []


def initialize():
Y
Yu Yang 已提交
23 24 25
    def convert_to_new_name(nm):
        return nm[:-len("_evaluator")]

Y
Yu Yang 已提交
26 27 28 29 30 31
    for __ev_name__ in filter(lambda x: x.endswith('_evaluator'), evs.__all__):
        __ev__ = getattr(evs, __ev_name__)
        if hasattr(__ev__, 'argspec'):
            argspec = __ev__.argspec
        else:
            argspec = inspect.getargspec(__ev__)
Y
Yu Yang 已提交
32 33
        parent_names = filter(lambda x: x in ['input', 'label', 'weight'],
                              argspec.args)
Y
Yu Yang 已提交
34 35 36 37 38
        v2_ev = __convert_to_v2__(
            __ev_name__,
            parent_names=parent_names,
            is_default_name='name' in argspec.args,
            attach_parent=True)
Y
Yu Yang 已提交
39 40 41 42 43 44

        __new_name__ = convert_to_new_name(__ev_name__)

        globals()[__new_name__] = v2_ev
        globals()[__new_name__].__name__ = __new_name__
        __all__.append(__new_name__)
Y
Yu Yang 已提交
45 46 47


initialize()