learn_subgraph_depths.cpp

Go to the documentation of this file.

/***************************************************************************
*                                                                          *
*   Copyright (C) 2018 by David B. Blumenthal                              *
*                                                                          *
*   This file is part of GEDLIB.                                           *
*                                                                          *
*   GEDLIB is free software: you can redistribute it and/or modify it      *
*   under the terms of the GNU Lesser General Public License as published  *
*   by the Free Software Foundation, either version 3 of the License, or   *
*   (at your option) any later version.                                    *
*                                                                          *
*   GEDLIB is distributed in the hope that it will be useful,              *
*   but WITHOUT ANY WARRANTY; without even the implied warranty of         *
*   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the           *
*   GNU Lesser General Public License for more details.                    *
*                                                                          *
*   You should have received a copy of the GNU Lesser General Public       *
*   License along with GEDLIB. If not, see <http://www.gnu.org/licenses/>. *
*                                                                          *
***************************************************************************/

#define GXL_GEDLIB_SHARED
#include "../../../src/env/ged_env.hpp"

void learn_depth_for_dataset(const std::string & dataset) {

    // Initialize environment.
    std::cout << "\n=== " << dataset << " ===\n";
    std::cout << "\tInitializing the environment ...\n";
    ged::GEDEnv<ged::GXLNodeID, ged::GXLLabel, ged::GXLLabel> env;
    std::vector<ged::GEDGraph::GraphID> graph_ids(env.load_gxl_graphs(std::string("../../../data/datasets/") + dataset + "/", std::string("../collections/") + dataset + "_50.xml"));
    if (dataset == "GREC") {
        env.set_edit_costs(ged::Options::EditCosts::GREC_1);
    }
    else {
        env.set_edit_costs(ged::Options::EditCosts::CHEM_1);
    }
    env.init();

    // Initialize the method.
    env.set_method(ged::Options::GEDMethod::SUBGRAPH, std::string("--threads 11 --save ../output/") + dataset + "_subgraph.ini");
    env.init_method();
}

int main(int argc, char* argv[]) {
    std::vector<std::string> datasets{"mao","pah","alkane","acyclic"};
    for (auto dataset : datasets) {
        try {
            learn_depth_for_dataset(dataset);
        }
        catch (const std::exception & error) {
            std::cerr << error.what() << ". " << "Error on " << dataset << "\n";
        }
    }
    return 0;
}