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()
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()