def main(args): data_loader = DataLoader() data_loader.load_graph() data_loader.load_sequence(args.seq_file) data_loader.construct_distribution() data_loader.generate_positive_samples() # data_loader.generate_negative_samples(3,1) model = SkipGram() model.initialize(data_loader.get_node_size(), args.emb_dim) model.train_process(args.epochs, data_loader, args.neg_num)