Function _create_option_parser

demo/hit_rate.py:28–55

Creates an option parser instance to handle command-line options.

()

def _create_option_parser():
    """ Creates an option parser instance to handle command-line options. """

    parser = argparse.ArgumentParser()
    parser.add_argument("--query_embed",
                        type=str,
                        required=True,
                        help='query embedding file with clicked item id')
    parser.add_argument("--item_embed",
                        type=str,
                        required=True,
                        help='item embedding file with item id')
    parser.add_argument("--topk",
                        type=str,
                        required=True,
                        default="100,200,500",
                        help='topK threshold list')
    parser.add_argument("--norm_item",
                        type=bool,
                        required=False,
                        default=False,
                        help='normalize item embedding')
    parser.add_argument("--norm_query",
                        type=bool,
                        required=False,
                        default=False,
                        help='normalize query embedding')
    return parser


def _yield_file(path):