BAnet/Time_BA.py at master · SinLaber/BAnet · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
import random
import networkx as nx
import matplotlib.pyplot as plt


def ba_bip_graph(n):
    g = nx.DiGraph()
    f_st = open("start.txt", 'r')
    line = f_st.readline()
    repeated_user_nodes = []
    repeated_item_nodes = []
    while line:
        repeated_user_nodes.append(line.strip('\n').split(" ")[0])
        repeated_item_nodes.append(line.strip('\n').split(" ")[1])
        line = f_st.readline()
    f_st.close()
    users = 0
    user_line = 0
    item_line = 0
    while users < n:
        m1 = random.randint(1, 10)                                   #新增一个user，添加随机0-9个边
        # print("m1:", m1)
        item_target = set()
        while len(item_target) < m1:
            x = random.choice(repeated_item_nodes)
            item_target.add(x)
        label = []
        f_u = open("user_name.txt", "r")
        line = f_u.readline()
        while line:
            label.append(line.strip('\n').split(" ")[0])
            line = f_u.readline()
        f_u.close()
        # print([label[user_line]] * m1)
        # print(item_target)
        r = zip([label[user_line]]*m1, item_target)
        for each in r:
            w = random.randint(1, 10)
            # g.add_edge(each[0], each[1], time=w)
            g.add_weighted_edges_from([(each[0], each[1], w)])
        repeated_item_nodes.extend(item_target)
        repeated_user_nodes.extend([label[user_line]]*m1)
        user_line += 1
        #新增一个user，新增4个item
        repeat = 0
        while repeat < 4:
            m2 = random.randint(1, 5)
            # print("m2:", m2)
            user_target = set()
            while len(user_target) < m2:
                x = random.choice(repeated_user_nodes)
                user_target.add(x)
            label = []
            f_i = open("item_name.txt", "r")
            line = f_i.readline()
            while line:
                label.append(line.strip('\n').split(" ")[0])
                line = f_i.readline()
            f_i.close()
            r = zip(user_target, [label[item_line]]*m2)
            for each in r:
                w = random.randint(1, 10)
                # g.add_edge(each[0], each[1], time=w)
                g.add_weighted_edges_from([(each[0], each[1], w)])
            # g.add_edges_from(zip([label[item_line]]*m2, user_target))
            repeated_user_nodes.extend(user_target)
            repeated_item_nodes.extend([label[item_line]] * m2)
            item_line += 1
            repeat += 1
        users = users+1
    return g


BA = ba_bip_graph(1000000)
pos = nx.circular_layout(BA)
nx.draw(BA, pos, with_labels=True, node_size=20, rotate=True)
plt.show()
f = open("BA_1000000.txt", "a")
# f.write('user\titem\ttime\n')
for edge in BA.edges(data=True):
    # print(edge)
    for key, value in BA.get_edge_data(edge[0], edge[1]).items():
        f.write(edge[0]+'\t'+edge[1]+'\t'+str(value)+'\n')
f.close()