diff --git a/toolkit.py b/toolkit.py index 80bd0dbd5a6036f8a079c07d87572920396c05b0..a318c21a6c58cda044e66d19edd68cf3665ddd5c 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__":