convertLGBMToRootTree.py File Reference

Go to the source code of this file.

Classes
class	util.convertLGBMToRootTree.LGBMTextNode

Namespaces
	util.convertLGBMToRootTree

Functions
def	util.convertLGBMToRootTree.lgbm_rawresponse_each_tree (model, my_input)
def	util.convertLGBMToRootTree.list2stdvector (values, dtype="float")
def	util.convertLGBMToRootTree.dump_tree (tree_structure)
def	util.convertLGBMToRootTree.dump2ROOT (model, output_filename, output_treename="lgbm")
def	util.convertLGBMToRootTree.convertLGBMToRootTree (model, output_filename, tree_name="lgbm")
def	util.convertLGBMToRootTree.test (model_file, tree_file, tree_name="lgbm", ntests=10000, test_file=None)
def	util.convertLGBMToRootTree.get_test_data (feature_names, test_file=None, ntests=None)
def	util.convertLGBMToRootTree.test_generic (booster, mvautils_predict, mva_utils, data_input)
def	util.convertLGBMToRootTree._ff (tree, node_infos)
def	util.convertLGBMToRootTree.test_detail_event (booster, mva_utils, input_values)
def	util.convertLGBMToRootTree.test_regression (booster, mva_utils, ntests=None, test_file=None)
def	util.convertLGBMToRootTree.test_binary (booster, mva_utils, ntests=None, test_file=None)
def	util.convertLGBMToRootTree.test_multiclass (booster, mva_utils, ntests=10000, test_file=None)
def	util.convertLGBMToRootTree.check_file (fn)

Variables
string	util.convertLGBMToRootTree.__doc__ = "Convert LightGBM model to TTree to be used with MVAUtils."
string	util.convertLGBMToRootTree.__author__ = "Ruggero Turra"
	util.convertLGBMToRootTree.level
	util.convertLGBMToRootTree.parser = argparse.ArgumentParser(description=__doc__)
	util.convertLGBMToRootTree.help
	util.convertLGBMToRootTree.nargs
	util.convertLGBMToRootTree.default
	util.convertLGBMToRootTree.action
	util.convertLGBMToRootTree.type
	util.convertLGBMToRootTree.int
	util.convertLGBMToRootTree.args = parser.parse_args()
	util.convertLGBMToRootTree.output
def	util.convertLGBMToRootTree.output_treename = convertLGBMToRootTree(args.input, args.output, args.tree_name)
def	util.convertLGBMToRootTree.result = test(args.input, args.output, args.tree_name, args.ntests, args.test_file)
	util.convertLGBMToRootTree.booster = lgb.Booster(model_file=args.input)
	util.convertLGBMToRootTree.objective = booster.dump_model()["objective"]