sklearn_utils.py 文件源码

def serialise_to_json(obj):
    """

    :param model:
    :return:
    """

    if obj is dict:
        return serialise_dict(obj)

    params = {}

    for kk, vv in obj.__dict__.iteritems():
        try:
            if isinstance(vv, type):
                continue

            if vv is None:
                params[kk] = None

            elif type(vv).__module__ == numpy_name:
                params[kk] = vv.tolist()

            elif isinstance(vv, dict):
                params[kk] = serialise_dict(vv)

            elif isinstance(vv, (str, list, bool, int, float)):
                params[kk] = vv

        except Exception as ex:
            raise ex

    return params