def METHOD_NAME(num_nodes,
                        neg_num,
                        embed_size,
                        sparse=False,
                        sparse_embedding=False,
                        shared_embedding=False):
    src = paddle.static.data("src", shape=[-1, 1], dtype="int64")
    dsts = paddle.static.data("dsts", shape=[-1, neg_num + 1], dtype="int64")
    py_reader = paddle.io.DataLoader.from_generator(
        capacity=64,
        feed_list=[src, dsts],
        iterable=False,
        use_double_buffer=False)
    model = SkipGramModel(
        num_nodes,
        embed_size,
        neg_num,
        sparse=sparse,
        sparse_embedding=sparse_embedding,
        shared_embedding=shared_embedding)
    loss = model(src, dsts)
    return py_reader, loss