/
main.cpp
43 lines (32 loc) · 1.26 KB
/
main.cpp
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
// Author: Anirudh Dagar 10/03/20
#include <iostream>
#include <fstream>
#include <string>
#include <sstream>
#include <Eigen>
#include "GCN_Model.h"
#include "data.h"
int main()
{
// Import Karate Club Data
auto adj_karate = read_karate();
// Import Weights
std::string address = "resources/saved/saved_weights.txt";
std::vector<Eigen::MatrixXd> weight_vec = getWeights(address);
std::cout<<"Imported PyTorch Trained Weights"<<std::endl;
int num_nodes = adj_karate.rows();
std::cout<<"num_nodes: "<<num_nodes<<std::endl;
Eigen::MatrixXd feats = Eigen::MatrixXd::Identity(num_nodes, num_nodes);
std::cout<<"Created feature matrix `X` or `feats` as Identity"<<std::endl;
int n_feat = feats.rows();
int n_hid = 10;
int n_out = 2;
std::cout<<"n_feat:"<<n_feat<<", n_hid:"<<n_hid<<", n_out:"<<n_out<<std::endl<<"\n";
std::cout<<"===============================================================\n";
GCN model(adj_karate, n_feat, n_hid, n_out, weight_vec);
Eigen::MatrixXd predictions = model.forward(feats);
std::cout<<"\nFinal Predictions:\n"<<predictions<<std::endl;
writeTofile("resources/saved/cpp_predicted.txt", Eigen::MatrixXd (predictions));
std::cout<<"Final Predictions saved!\n";
return 0;
}