static void gd_begin_graph_to_file(graph_t* g, box bb, point pb)
{
char *truecolor_p;
gd_begin_graph(g, bb, pb);
if (Verbose)
fprintf(stderr,"%s: allocating a %dK GD image\n",
CmdName, ROUND( Viewport.x * Viewport.y / 1024. ));
truecolor_p = agget(g,"truecolor");
/* automatically decide color mode if "truecolor" not specified */
if (truecolor_p == NULL || *truecolor_p == '\0') {
truecolor_p="false"; /* indexed color maps unless ... */
if (agfindattr(g->proto->n, "shapefile")) {
/* ... custom shapefiles are used */
truecolor_p = "true";
}
}
if (mapbool(truecolor_p))
im = gdImageCreateTrueColor(Viewport.x, Viewport.y);
else
im = gdImageCreate(Viewport.x, Viewport.y);
init_gd();
#ifdef MYTRACE
fprintf(stderr,"gd_begin_graph_to_file\n");
#endif
}
static void gd_begin_graph_to_memory(graph_t* g, box bb, point pb)
{
gd_begin_graph(g, bb, pb);
if (Verbose)
fprintf(stderr,"%s: using existing GD image\n", CmdName);
im = *(gdImagePtr *)Output_file;
init_gd();
#ifdef MYTRACE
fprintf(stderr,"gd_begin_graph_to_memory\n");
#endif
}
int main(int argc, char* argv[]) {
if (!config(argc, argv)) {
return 0;
}
/* init N, K, T */
read_csv_graph(&"../../data/jsim_graph_selected/"[0u]);
init();
/* train */
if (option == 1) {
for (int t = start_T; t < T; t++) {
cout << "t = " << t << endl;
init_gd(t);
train_gd(t, stepsize, delta, lambda);
}
}
else if (option == 2) {
for (int t = start_T; t < T; t++) {
cout << "t = " << t << endl;
init_em(t);
train_em(t, stepsize, delta);
}
}
cout << "training done!" << endl;
/* save */
if (option == 1) {
char* c_likel = "likelihood.txt";
// output_hidden(&"../../data/dict/user_id_map.dat"[0u], &"./save/baseline_0/"[0u]);
// output_1d(&"./save/baseline_0/likelihood.txt"[0u], likel, T);
output_hidden(&"../../data/dict/user_id_map.dat"[0u], outdir);
output_1d(strcat(outdir, c_likel), likel, T); // likelihood
}
else if (option == 2) {
char* c_alpha = "alpha.txt";
char* c_likel = "likelihood.txt";
// output_hidden(&"../../data/dict/user_id_map.dat"[0u], &"./save/lfpm/"[0u]);
// output_1d(&"./save/lfpm/alpha.txt"[0u], alpha_s, T);
// output_1d(&"./save/lfpm/likelihood.txt"[0u], likel, T);
output_hidden(&"../../data/dict/user_id_map.dat"[0u], outdir);
output_1d(strcat(outdir, c_alpha), alpha_s, T);
output_1d(strcat(outdir, c_likel), likel, T);
}
}
