Metadata-Version: 2.1
Name: automl-lib
Version: 0.0.1
Summary: automl fm dnn deepFM in config file
Home-page: UNKNOWN
Author: pig_zhu
Author-email: zhipeng875820@163.com
License: UNKNOWN
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.6
Description-Content-Type: text/markdown
Requires-Dist: tensorflow (==1.15.0)
Requires-Dist: pyyaml
Requires-Dist: sklearn


# 使用方法，对外提供两个方法build_framework，export_saved_model
'''
# -*- coding: utf-8 -*-
# @Author : Zip
# @Time   : 2020/12/18|12:53
# @Moto   : Knowledge comes from decomposition
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import warnings
from lib.estimator import build_framework
from lib.export import export_saved_model
import argparse

warnings.filterwarnings('ignore')


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--config_file", default="conf/job_sort_with_fm_with_two_feature.yaml")
    args = parser.parse_args()

    build_framework(args.config_file)
    # export_saved_model(args.config_file)


if __name__ == '__main__':
    main()


'''

