tensor-sc/d3c_cocluster.cpp at master · arbenson/tensor-sc · 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
#include "common.hpp"
#include "community.hpp"
#include "cut_tools.hpp"
#include "file_io.hpp"
#include "network.hpp"

#include <iostream>

int main(int argc, char **argv) {
    if (argc == 1) {
	PrintHelp();
	return 0;
    }
    int network = atoi(argv[1]);
    int num_comms = 3;
    int triple_type = D3CS;
    int cut_type = D3C_COND;
    std::string name = NetworkStr(network);

    {
	Network net = GetNetwork(network, triple_type);
	Cutter cutter(net, 0, cut_type, name + "_cocluster_u");

	std::string file = "data/" + name + "_u";
	std::vector<double> vals = ReadVector<double>(file);
	Vector<double> u_vals(vals.size());
	for (int i = 0; i < vals.size(); ++i) {
	    u_vals(i) = vals[i];
	}

	std::vector<int> order = GetSortedOrder(u_vals);
	WriteVector(order, name + "_cocluster_u_order.txt");
	//std::vector<int> cut = cutter.GetCut(order);
    }

    {
	Network net = GetNetwork(network, triple_type);
	Cutter cutter(net, 0, cut_type, name + "_cocluster_v");

	std::string file = "data/" + name + "_v";
	std::vector<double> vals = ReadVector<double>(file);
	Vector<double> v_vals(vals.size());
	for (int i = 0; i < vals.size(); ++i) {
	    v_vals(i) = vals[i];
	}

	std::vector<int> order = GetSortedOrder(v_vals);
	WriteVector(order, name + "_cocluster_v_order.txt");
	//std::vector<int> cut = cutter.GetCut(order);
    }

    std::cout << "END ---------" << std::endl;
}