ATLAS Offline Software
Loading...
Searching...
No Matches
convertLGBMToRootTree.py File Reference

Go to the source code of this file.

Classes

class  util.convertLGBMToRootTree.LGBMTextNode

Namespaces

namespace  util
namespace  util.convertLGBMToRootTree

Functions

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

Variables

str 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
 util.convertLGBMToRootTree.output_treename = convertLGBMToRootTree(args.input, args.output, args.tree_name)
 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"]