From 0519ccd89498e8f56d61d2bbb3f65c7a77fe25e0 Mon Sep 17 00:00:00 2001
From: Nikolai <osterei33@gmx.de>
Date: Fri, 10 Aug 2018 11:35:29 +0200
Subject: [PATCH] starting initialisation from pandas df
---
toolkit.py | 21 ++++++++++++++-------
1 file changed, 14 insertions(+), 7 deletions(-)
diff --git a/toolkit.py b/toolkit.py
index 80bd0db..a318c21 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -1197,15 +1197,22 @@ for dataset_name in ClassificationProject.dataset_names:
create_setter(dataset_name)))
-# class ClassificationProjectDataFrame(ClassificationProject):
+class ClassificationProjectDataFrame(ClassificationProject):
-# """
-# A little hack to initialize a ClassificationProject from a pandas DataFrame instead of ROOT TTrees
-# """
+ """
+ A little hack to initialize a ClassificationProject from a pandas DataFrame instead of ROOT TTrees
+ """
-# def __init__(self,
-# **kwargs):
-# pass
+ def __init__(self,
+ input_columns,
+ weight_column="weights",
+ label_column="labels",
+ signal_label="signal",
+ background_label="background",
+ split_mode="split_column",
+ split_colurm="is_train",
+ **kwargs):
+ pass
if __name__ == "__main__":
--
GitLab