#!/usr/bin/env python
# Copyright (c) 2011-2012, Universite de Versailles St-Quentin-en-Yvelines
#
# This file is part of ASK.  ASK is free software: you can redistribute
# it and/or modify it under the terms of the GNU General Public
# License as published by the Free Software Foundation, version 2.
#
# This program 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 General Public License for more
# details.
#
# You should have received a copy of the GNU General Public License along with
# this program; if not, write to the Free Software Foundation, Inc., 51
# Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.


import common.cart
from common.configuration import Configuration
from common.tree import save_tree

if __name__ == "__main__":
    import argparse
    parser = argparse.ArgumentParser(description="Builds a cart tree model")
    parser.add_argument('configuration')
    parser.add_argument('labelled_file')
    parser.add_argument('output_file')
    args = parser.parse_args()

    conf = Configuration(args.configuration)
    T = common.cart.build_tree(conf, conf("modules.model.params.cp",
                                          float, 0.01),
                               args.labelled_file)
    save_tree(T, args.output_file)
