Jormungandr-code/streaming.py at main · LozzaTray/Jormungandr-code · 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
from data.twitch import TwitchGraph
from model.graph_mcmc import Graph_MCMC

def save():
    print("Loading data...")
    twitch = TwitchGraph()
    print("Converting to graph-tool...")
    graph = twitch.generate_mcmc_graph()
    print("Saving to gt file...")
    graph.save_to_file("twitch")
    print("Done")

def analyse():
    print("Loading from file...")
    graph = Graph_MCMC()
    graph.read_from_file("twitch.gt")
    print("Done")

    graph.partition(B_min=4, B_max=4)
    graph.mcmc(100, verbose=True)
    graph.draw("twitch.png")

    #graph.plot_posterior_props()
    names = graph.get_feature_names()

    classifier = graph.sample_classifier_mala(100, step_scaling=0.001, verbose=True)
    classifier.thin_samples()


    #classifier.plot_U()
    classifier.plot_block_principal_dims(names.copy(), 1)
    classifier.plot_sampled_weights(names.copy(), std_dev_multiplier=1, null_space=1)


if __name__ == "__main__":
    analyse()