示例#1
0
文件: main.py 项目: SongFGH/FSCNMF-2
def learn_model(args):
    """
    Method to create adjacency matrix powers, read features, and learn embedding.
    :param args: Arguments object.
    """
    A = read_graph(args.edge_path, args.order)
    X = read_features(args.feature_path)
    model = FSCNMF(A, X, args)
    model.optimize()
    model.save_embedding()
示例#2
0
文件: main.py 项目: zhanghegui/TENE
def learn_model(args):
    """
    Method to create adjacency matrix powers, read features, and learn embedding.
    :param args: Arguments object.
    """
    X = read_graph(args.edge_path, args.order)
    if args.features == "dense":
        T = read_features(args.feature_path)
    elif args.features == "sparse":
        T = read_sparse_features(args.feature_path)
    model = TENE(X, T, args)
    model.optimize()
    model.save_embedding()