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)
