Logo ROOT  

Reference Guide
 
Loading...
Searching...
No Matches
TMVA_CNN_Classification.C File Reference
Tutorials » TMVA tutorials

Detailed Description

View in nbviewer Open in SWAN
TMVA Classification Example Using a Convolutional Neural Network

This is an example of using a CNN in TMVA. We do classification using a toy image data set that is generated when running the example macro

Running with nthreads = 4
DataSetInfo : [dataset] : Added class "Signal"
: Add Tree sig_tree of type Signal with 1000 events
DataSetInfo : [dataset] : Added class "Background"
: Add Tree bkg_tree of type Background with 1000 events
Factory : Booking method: ␛[1mBDT␛[0m
:
: Rebuilding Dataset dataset
: Building event vectors for type 2 Signal
: Dataset[dataset] : create input formulas for tree sig_tree
: Using variable vars[0] from array expression vars of size 256
: Building event vectors for type 2 Background
: Dataset[dataset] : create input formulas for tree bkg_tree
: Using variable vars[0] from array expression vars of size 256
DataSetFactory : [dataset] : Number of events in input trees
:
:
: Number of training and testing events
: ---------------------------------------------------------------------------
: Signal -- training events : 800
: Signal -- testing events : 200
: Signal -- training and testing events: 1000
: Background -- training events : 800
: Background -- testing events : 200
: Background -- training and testing events: 1000
:
Factory : Booking method: ␛[1mTMVA_DNN_CPU␛[0m
:
: Parsing option string:
: ... "!H:V:ErrorStrategy=CROSSENTROPY:VarTransform=None:WeightInitialization=XAVIER:Layout=DENSE|100|RELU,BNORM,DENSE|100|RELU,BNORM,DENSE|100|RELU,BNORM,DENSE|100|RELU,DENSE|1|LINEAR:TrainingStrategy=LearningRate=1e-3,Momentum=0.9,Repetitions=1,ConvergenceSteps=5,BatchSize=100,TestRepetitions=1,MaxEpochs=10,WeightDecay=1e-4,Regularization=None,Optimizer=ADAM,DropConfig=0.0+0.0+0.0+0.:Architecture=CPU"
: The following options are set:
: - By User:
: <none>
: - Default:
: Boost_num: "0" [Number of times the classifier will be boosted]
: Parsing option string:
: ... "!H:V:ErrorStrategy=CROSSENTROPY:VarTransform=None:WeightInitialization=XAVIER:Layout=DENSE|100|RELU,BNORM,DENSE|100|RELU,BNORM,DENSE|100|RELU,BNORM,DENSE|100|RELU,DENSE|1|LINEAR:TrainingStrategy=LearningRate=1e-3,Momentum=0.9,Repetitions=1,ConvergenceSteps=5,BatchSize=100,TestRepetitions=1,MaxEpochs=10,WeightDecay=1e-4,Regularization=None,Optimizer=ADAM,DropConfig=0.0+0.0+0.0+0.:Architecture=CPU"
: The following options are set:
: - By User:
: V: "True" [Verbose output (short form of "VerbosityLevel" below - overrides the latter one)]
: VarTransform: "None" [List of variable transformations performed before training, e.g., "D_Background,P_Signal,G,N_AllClasses" for: "Decorrelation, PCA-transformation, Gaussianisation, Normalisation, each for the given class of events ('AllClasses' denotes all events of all classes, if no class indication is given, 'All' is assumed)"]
: H: "False" [Print method-specific help message]
: Layout: "DENSE|100|RELU,BNORM,DENSE|100|RELU,BNORM,DENSE|100|RELU,BNORM,DENSE|100|RELU,DENSE|1|LINEAR" [Layout of the network.]
: ErrorStrategy: "CROSSENTROPY" [Loss function: Mean squared error (regression) or cross entropy (binary classification).]
: WeightInitialization: "XAVIER" [Weight initialization strategy]
: Architecture: "CPU" [Which architecture to perform the training on.]
: TrainingStrategy: "LearningRate=1e-3,Momentum=0.9,Repetitions=1,ConvergenceSteps=5,BatchSize=100,TestRepetitions=1,MaxEpochs=10,WeightDecay=1e-4,Regularization=None,Optimizer=ADAM,DropConfig=0.0+0.0+0.0+0." [Defines the training strategies.]
: - Default:
: VerbosityLevel: "Default" [Verbosity level]
: CreateMVAPdfs: "False" [Create PDFs for classifier outputs (signal and background)]
: IgnoreNegWeightsInTraining: "False" [Events with negative weights are ignored in the training (but are included for testing and performance evaluation)]
: InputLayout: "0|0|0" [The Layout of the input]
: BatchLayout: "0|0|0" [The Layout of the batch]
: RandomSeed: "0" [Random seed used for weight initialization and batch shuffling]
: ValidationSize: "20%" [Part of the training data to use for validation. Specify as 0.2 or 20% to use a fifth of the data set as validation set. Specify as 100 to use exactly 100 events. (Default: 20%)]
: Will now use the CPU architecture with BLAS and IMT support !
Factory : Booking method: ␛[1mTMVA_CNN_CPU␛[0m
:
: Parsing option string:
: ... "!H:V:ErrorStrategy=CROSSENTROPY:VarTransform=None:WeightInitialization=XAVIER:InputLayout=1|16|16:Layout=CONV|10|3|3|1|1|1|1|RELU,BNORM,CONV|10|3|3|1|1|1|1|RELU,MAXPOOL|2|2|1|1,RESHAPE|FLAT,DENSE|100|RELU,DENSE|1|LINEAR:TrainingStrategy=LearningRate=1e-3,Momentum=0.9,Repetitions=1,ConvergenceSteps=5,BatchSize=100,TestRepetitions=1,MaxEpochs=10,WeightDecay=1e-4,Regularization=None,Optimizer=ADAM,DropConfig=0.0+0.0+0.0+0.0:Architecture=CPU"
: The following options are set:
: - By User:
: <none>
: - Default:
: Boost_num: "0" [Number of times the classifier will be boosted]
: Parsing option string:
: ... "!H:V:ErrorStrategy=CROSSENTROPY:VarTransform=None:WeightInitialization=XAVIER:InputLayout=1|16|16:Layout=CONV|10|3|3|1|1|1|1|RELU,BNORM,CONV|10|3|3|1|1|1|1|RELU,MAXPOOL|2|2|1|1,RESHAPE|FLAT,DENSE|100|RELU,DENSE|1|LINEAR:TrainingStrategy=LearningRate=1e-3,Momentum=0.9,Repetitions=1,ConvergenceSteps=5,BatchSize=100,TestRepetitions=1,MaxEpochs=10,WeightDecay=1e-4,Regularization=None,Optimizer=ADAM,DropConfig=0.0+0.0+0.0+0.0:Architecture=CPU"
: The following options are set:
: - By User:
: V: "True" [Verbose output (short form of "VerbosityLevel" below - overrides the latter one)]
: VarTransform: "None" [List of variable transformations performed before training, e.g., "D_Background,P_Signal,G,N_AllClasses" for: "Decorrelation, PCA-transformation, Gaussianisation, Normalisation, each for the given class of events ('AllClasses' denotes all events of all classes, if no class indication is given, 'All' is assumed)"]
: H: "False" [Print method-specific help message]
: InputLayout: "1|16|16" [The Layout of the input]
: Layout: "CONV|10|3|3|1|1|1|1|RELU,BNORM,CONV|10|3|3|1|1|1|1|RELU,MAXPOOL|2|2|1|1,RESHAPE|FLAT,DENSE|100|RELU,DENSE|1|LINEAR" [Layout of the network.]
: ErrorStrategy: "CROSSENTROPY" [Loss function: Mean squared error (regression) or cross entropy (binary classification).]
: WeightInitialization: "XAVIER" [Weight initialization strategy]
: Architecture: "CPU" [Which architecture to perform the training on.]
: TrainingStrategy: "LearningRate=1e-3,Momentum=0.9,Repetitions=1,ConvergenceSteps=5,BatchSize=100,TestRepetitions=1,MaxEpochs=10,WeightDecay=1e-4,Regularization=None,Optimizer=ADAM,DropConfig=0.0+0.0+0.0+0.0" [Defines the training strategies.]
: - Default:
: VerbosityLevel: "Default" [Verbosity level]
: CreateMVAPdfs: "False" [Create PDFs for classifier outputs (signal and background)]
: IgnoreNegWeightsInTraining: "False" [Events with negative weights are ignored in the training (but are included for testing and performance evaluation)]
: BatchLayout: "0|0|0" [The Layout of the batch]
: RandomSeed: "0" [Random seed used for weight initialization and batch shuffling]
: ValidationSize: "20%" [Part of the training data to use for validation. Specify as 0.2 or 20% to use a fifth of the data set as validation set. Specify as 100 to use exactly 100 events. (Default: 20%)]
: Will now use the CPU architecture with BLAS and IMT support !
Model: "sequential"
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
reshape (Reshape) (None, 16, 16, 1) 0
conv2d (Conv2D) (None, 16, 16, 10) 100
batch_normalization (Batch (None, 16, 16, 10) 40
Normalization)
conv2d_1 (Conv2D) (None, 16, 16, 10) 910
max_pooling2d (MaxPooling2 (None, 15, 15, 10) 0
D)
flatten (Flatten) (None, 2250) 0
dense (Dense) (None, 256) 576256
dense_1 (Dense) (None, 2) 514
=================================================================
Total params: 577820 (2.20 MB)
Trainable params: 577800 (2.20 MB)
Non-trainable params: 20 (80.00 Byte)
_________________________________________________________________
Factory : Booking method: ␛[1mPyKeras␛[0m
:
: Setting up tf.keras
: Using TensorFlow version 2
: Use Keras version from TensorFlow : tf.keras
: Applying GPU option: gpu_options.allow_growth=True
: Loading Keras Model
: Loaded model from file: model_cnn.h5
Factory : Booking method: ␛[1mPyTorch␛[0m
:
: Using PyTorch - setting special configuration options
: Using PyTorch version 2
: Setup PyTorch Model for training
: Executing user initialization code from /home/sftnight/build/workspace/root-makedoc-v628/rootspi/rdoc/src/v6-28-00-patches.build/tutorials/tmva/PyTorch_Generate_CNN_Model.py
running Torch code defining the model....
The PyTorch CNN model is created and saved as PyTorchModelCNN.pt
: Loaded pytorch train function:
: Loaded pytorch optimizer:
: Loaded pytorch loss function:
: Loaded pytorch predict function:
: Loaded model from file: PyTorchModelCNN.pt
Factory : ␛[1mTrain all methods␛[0m
Factory : Train method: BDT for Classification
:
BDT : #events: (reweighted) sig: 800 bkg: 800
: #events: (unweighted) sig: 800 bkg: 800
: Training 200 Decision Trees ... patience please
: Elapsed time for training with 1600 events: 0.955 sec
BDT : [dataset] : Evaluation of BDT on training sample (1600 events)
: Elapsed time for evaluation of 1600 events: 0.0272 sec
: Creating xml weight file: ␛[0;36mdataset/weights/TMVA_CNN_Classification_BDT.weights.xml␛[0m
: Creating standalone class: ␛[0;36mdataset/weights/TMVA_CNN_Classification_BDT.class.C␛[0m
: TMVA_CNN_ClassificationOutput.root:/dataset/Method_BDT/BDT
Factory : Training finished
:
Factory : Train method: TMVA_DNN_CPU for Classification
:
: Start of deep neural network training on CPU using MT, nthreads = 4
:
: ***** Deep Learning Network *****
DEEP NEURAL NETWORK: Depth = 8 Input = ( 1, 1, 256 ) Batch size = 100 Loss function = C
Layer 0 DENSE Layer: ( Input = 256 , Width = 100 ) Output = ( 1 , 100 , 100 ) Activation Function = Relu
Layer 1 BATCH NORM Layer: Input/Output = ( 100 , 100 , 1 ) Norm dim = 100 axis = -1
Layer 2 DENSE Layer: ( Input = 100 , Width = 100 ) Output = ( 1 , 100 , 100 ) Activation Function = Relu
Layer 3 BATCH NORM Layer: Input/Output = ( 100 , 100 , 1 ) Norm dim = 100 axis = -1
Layer 4 DENSE Layer: ( Input = 100 , Width = 100 ) Output = ( 1 , 100 , 100 ) Activation Function = Relu
Layer 5 BATCH NORM Layer: Input/Output = ( 100 , 100 , 1 ) Norm dim = 100 axis = -1
Layer 6 DENSE Layer: ( Input = 100 , Width = 100 ) Output = ( 1 , 100 , 100 ) Activation Function = Relu
Layer 7 DENSE Layer: ( Input = 100 , Width = 1 ) Output = ( 1 , 100 , 1 ) Activation Function = Identity
: Using 1280 events for training and 320 for testing
: Compute initial loss on the validation data
: Training phase 1 of 1: Optimizer ADAM (beta1=0.9,beta2=0.999,eps=1e-07) Learning rate = 0.001 regularization 0 minimum error = inf
: --------------------------------------------------------------
: Epoch | Train Err. Val. Err. t(s)/epoch t(s)/Loss nEvents/s Conv. Steps
: --------------------------------------------------------------
: Start epoch iteration ...
: 1 Minimum Test error found - save the configuration
: 1 | 1.01956 0.867828 0.184791 0.0162643 7120.55 0
: 2 Minimum Test error found - save the configuration
: 2 | 0.687791 0.721726 0.183275 0.0161699 7181.1 0
: 3 Minimum Test error found - save the configuration
: 3 | 0.592872 0.67376 0.182098 0.0159038 7220.48 0
: 4 Minimum Test error found - save the configuration
: 4 | 0.542811 0.672206 0.182866 0.0158263 7183.93 0
: 5 Minimum Test error found - save the configuration
: 5 | 0.465044 0.652241 0.183508 0.0159054 7159.78 0
: 6 Minimum Test error found - save the configuration
: 6 | 0.418565 0.642508 0.181728 0.015877 7235.39 0
: 7 | 0.357129 0.678295 0.182552 0.0152313 7171.85 1
: 8 | 0.314569 0.692079 0.185016 0.0152121 7066.98 2
: 9 | 0.289278 0.716658 0.182462 0.0152186 7175.16 3
: 10 | 0.257828 0.698478 0.182195 0.0150736 7180.43 4
:
: Elapsed time for training with 1600 events: 1.87 sec
: Evaluate deep neural network on CPU using batches with size = 100
:
TMVA_DNN_CPU : [dataset] : Evaluation of TMVA_DNN_CPU on training sample (1600 events)
: Elapsed time for evaluation of 1600 events: 0.084 sec
: Creating xml weight file: ␛[0;36mdataset/weights/TMVA_CNN_Classification_TMVA_DNN_CPU.weights.xml␛[0m
: Creating standalone class: ␛[0;36mdataset/weights/TMVA_CNN_Classification_TMVA_DNN_CPU.class.C␛[0m
Factory : Training finished
:
Factory : Train method: TMVA_CNN_CPU for Classification
:
: Start of deep neural network training on CPU using MT, nthreads = 4
:
: ***** Deep Learning Network *****
DEEP NEURAL NETWORK: Depth = 7 Input = ( 1, 16, 16 ) Batch size = 100 Loss function = C
Layer 0 CONV LAYER: ( W = 16 , H = 16 , D = 10 ) Filter ( W = 3 , H = 3 ) Output = ( 100 , 10 , 10 , 256 ) Activation Function = Relu
Layer 1 BATCH NORM Layer: Input/Output = ( 10 , 256 , 100 ) Norm dim = 10 axis = 1
Layer 2 CONV LAYER: ( W = 16 , H = 16 , D = 10 ) Filter ( W = 3 , H = 3 ) Output = ( 100 , 10 , 10 , 256 ) Activation Function = Relu
Layer 3 POOL Layer: ( W = 15 , H = 15 , D = 10 ) Filter ( W = 2 , H = 2 ) Output = ( 100 , 10 , 10 , 225 )
Layer 4 RESHAPE Layer Input = ( 10 , 15 , 15 ) Output = ( 1 , 100 , 2250 )
Layer 5 DENSE Layer: ( Input = 2250 , Width = 100 ) Output = ( 1 , 100 , 100 ) Activation Function = Relu
Layer 6 DENSE Layer: ( Input = 100 , Width = 1 ) Output = ( 1 , 100 , 1 ) Activation Function = Identity
: Using 1280 events for training and 320 for testing
: Compute initial loss on the validation data
: Training phase 1 of 1: Optimizer ADAM (beta1=0.9,beta2=0.999,eps=1e-07) Learning rate = 0.001 regularization 0 minimum error = inf
: --------------------------------------------------------------
: Epoch | Train Err. Val. Err. t(s)/epoch t(s)/Loss nEvents/s Conv. Steps
: --------------------------------------------------------------
: Start epoch iteration ...
: 1 Minimum Test error found - save the configuration
: 1 | 0.943272 0.73297 1.49509 0.113369 868.481 0
: 2 Minimum Test error found - save the configuration
: 2 | 0.702442 0.697409 1.46943 0.112648 884.447 0
: 3 | 0.669219 0.721644 1.46099 0.111632 889.311 1
: 4 Minimum Test error found - save the configuration
: 4 | 0.652187 0.69542 1.46297 0.112927 888.858 0
: 5 Minimum Test error found - save the configuration
: 5 | 0.619355 0.645451 1.4808 0.117081 879.948 0
: 6 Minimum Test error found - save the configuration
: 6 | 0.577407 0.633669 1.45266 0.112389 895.344 0
: 7 | 0.548875 0.664203 1.46651 0.111687 885.724 1
: 8 Minimum Test error found - save the configuration
: 8 | 0.503401 0.582234 1.47336 0.111286 881.007 0
: 9 Minimum Test error found - save the configuration
: 9 | 0.442678 0.562936 1.45243 0.112145 895.334 0
: 10 | 0.393351 0.583643 1.45646 0.11157 892.269 1
:
: Elapsed time for training with 1600 events: 14.8 sec
: Evaluate deep neural network on CPU using batches with size = 100
:
TMVA_CNN_CPU : [dataset] : Evaluation of TMVA_CNN_CPU on training sample (1600 events)
: Elapsed time for evaluation of 1600 events: 0.594 sec
: Creating xml weight file: ␛[0;36mdataset/weights/TMVA_CNN_Classification_TMVA_CNN_CPU.weights.xml␛[0m
: Creating standalone class: ␛[0;36mdataset/weights/TMVA_CNN_Classification_TMVA_CNN_CPU.class.C␛[0m
Factory : Training finished
:
Factory : Train method: PyKeras for Classification
:
:
: ␛[1m================================================================␛[0m
: ␛[1mH e l p f o r M V A m e t h o d [ PyKeras ] :␛[0m
:
: Keras is a high-level API for the Theano and Tensorflow packages.
: This method wraps the training and predictions steps of the Keras
: Python package for TMVA, so that dataloading, preprocessing and
: evaluation can be done within the TMVA system. To use this Keras
: interface, you have to generate a model with Keras first. Then,
: this model can be loaded and trained in TMVA.
:
:
: <Suppress this message by specifying "!H" in the booking option>
: ␛[1m================================================================␛[0m
:
: Split TMVA training data in 1280 training events and 320 validation events
: Training Model Summary
custom objects for loading model : {'optimizer': <class 'torch.optim.adam.Adam'>, 'criterion': BCELoss(), 'train_func': <function fit at 0x7f3954d5fca0>, 'predict_func': <function predict at 0x7f3954d5fdc0>}
Model: "sequential"
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
reshape (Reshape) (None, 16, 16, 1) 0
conv2d (Conv2D) (None, 16, 16, 10) 100
batch_normalization (Batch (None, 16, 16, 10) 40
Normalization)
conv2d_1 (Conv2D) (None, 16, 16, 10) 910
max_pooling2d (MaxPooling2 (None, 15, 15, 10) 0
D)
flatten (Flatten) (None, 2250) 0
dense (Dense) (None, 256) 576256
dense_1 (Dense) (None, 2) 514
=================================================================
Total params: 577820 (2.20 MB)
Trainable params: 577800 (2.20 MB)
Non-trainable params: 20 (80.00 Byte)
_________________________________________________________________
: Option SaveBestOnly: Only model weights with smallest validation loss will be stored
Epoch 1/10
1/13 [=>............................] - ETA: 10s - loss: 0.8101 - accuracy: 0.5600␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
3/13 [=====>........................] - ETA: 0s - loss: 2.1564 - accuracy: 0.5300 ␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
6/13 [============>.................] - ETA: 0s - loss: 1.7482 - accuracy: 0.5200␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
9/13 [===================>..........] - ETA: 0s - loss: 1.5394 - accuracy: 0.5022␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
12/13 [==========================>...] - ETA: 0s - loss: 1.3623 - accuracy: 0.5075
Epoch 1: val_loss improved from inf to 1.49201, saving model to trained_model_cnn.h5
␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
13/13 [==============================] - 2s 59ms/step - loss: 1.3246 - accuracy: 0.5070 - val_loss: 1.4920 - val_accuracy: 0.5281
Epoch 2/10
1/13 [=>............................] - ETA: 0s - loss: 0.7943 - accuracy: 0.4800␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
4/13 [========>.....................] - ETA: 0s - loss: 0.7345 - accuracy: 0.5425␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
7/13 [===============>..............] - ETA: 0s - loss: 0.7228 - accuracy: 0.5257␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
10/13 [======================>.......] - ETA: 0s - loss: 0.7077 - accuracy: 0.5450␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
13/13 [==============================] - ETA: 0s - loss: 0.7028 - accuracy: 0.5492
Epoch 2: val_loss improved from 1.49201 to 0.79314, saving model to trained_model_cnn.h5
␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
13/13 [==============================] - 0s 29ms/step - loss: 0.7028 - accuracy: 0.5492 - val_loss: 0.7931 - val_accuracy: 0.6125
Epoch 3/10
1/13 [=>............................] - ETA: 0s - loss: 0.6482 - accuracy: 0.7200␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
4/13 [========>.....................] - ETA: 0s - loss: 0.6541 - accuracy: 0.6525␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
7/13 [===============>..............] - ETA: 0s - loss: 0.6641 - accuracy: 0.6214␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
10/13 [======================>.......] - ETA: 0s - loss: 0.6641 - accuracy: 0.6160␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
13/13 [==============================] - ETA: 0s - loss: 0.6655 - accuracy: 0.6078
Epoch 3: val_loss improved from 0.79314 to 0.67703, saving model to trained_model_cnn.h5
␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
13/13 [==============================] - 0s 29ms/step - loss: 0.6655 - accuracy: 0.6078 - val_loss: 0.6770 - val_accuracy: 0.5875
Epoch 4/10
1/13 [=>............................] - ETA: 0s - loss: 0.6440 - accuracy: 0.6400␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
4/13 [========>.....................] - ETA: 0s - loss: 0.6490 - accuracy: 0.6175␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
7/13 [===============>..............] - ETA: 0s - loss: 0.6485 - accuracy: 0.6143␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
10/13 [======================>.......] - ETA: 0s - loss: 0.6521 - accuracy: 0.6090␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
13/13 [==============================] - ETA: 0s - loss: 0.6468 - accuracy: 0.6172
Epoch 4: val_loss did not improve from 0.67703
␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
13/13 [==============================] - 0s 23ms/step - loss: 0.6468 - accuracy: 0.6172 - val_loss: 0.8475 - val_accuracy: 0.5281
Epoch 5/10
1/13 [=>............................] - ETA: 0s - loss: 0.6420 - accuracy: 0.5700␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
4/13 [========>.....................] - ETA: 0s - loss: 0.6456 - accuracy: 0.5975␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
7/13 [===============>..............] - ETA: 0s - loss: 0.6362 - accuracy: 0.6200␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
11/13 [========================>.....] - ETA: 0s - loss: 0.6364 - accuracy: 0.6273
Epoch 5: val_loss improved from 0.67703 to 0.65620, saving model to trained_model_cnn.h5
␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
13/13 [==============================] - 0s 25ms/step - loss: 0.6317 - accuracy: 0.6414 - val_loss: 0.6562 - val_accuracy: 0.6500
Epoch 6/10
1/13 [=>............................] - ETA: 0s - loss: 0.5917 - accuracy: 0.7300␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
5/13 [==========>...................] - ETA: 0s - loss: 0.5992 - accuracy: 0.7160␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
8/13 [=================>............] - ETA: 0s - loss: 0.6098 - accuracy: 0.7025␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
11/13 [========================>.....] - ETA: 0s - loss: 0.6042 - accuracy: 0.7045
Epoch 6: val_loss did not improve from 0.65620
␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
13/13 [==============================] - 0s 22ms/step - loss: 0.6068 - accuracy: 0.6977 - val_loss: 0.6760 - val_accuracy: 0.5875
Epoch 7/10
1/13 [=>............................] - ETA: 0s - loss: 0.5609 - accuracy: 0.7900␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
5/13 [==========>...................] - ETA: 0s - loss: 0.5689 - accuracy: 0.7420␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
9/13 [===================>..........] - ETA: 0s - loss: 0.5901 - accuracy: 0.7011␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
12/13 [==========================>...] - ETA: 0s - loss: 0.5963 - accuracy: 0.6842
Epoch 7: val_loss improved from 0.65620 to 0.65189, saving model to trained_model_cnn.h5
␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
13/13 [==============================] - 0s 30ms/step - loss: 0.5947 - accuracy: 0.6867 - val_loss: 0.6519 - val_accuracy: 0.6344
Epoch 8/10
1/13 [=>............................] - ETA: 0s - loss: 0.5555 - accuracy: 0.7100␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
4/13 [========>.....................] - ETA: 0s - loss: 0.5468 - accuracy: 0.7350␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
7/13 [===============>..............] - ETA: 0s - loss: 0.5508 - accuracy: 0.7443␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
11/13 [========================>.....] - ETA: 0s - loss: 0.5555 - accuracy: 0.7482
Epoch 8: val_loss did not improve from 0.65189
␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
13/13 [==============================] - 0s 23ms/step - loss: 0.5557 - accuracy: 0.7492 - val_loss: 0.6662 - val_accuracy: 0.6094
Epoch 9/10
1/13 [=>............................] - ETA: 0s - loss: 0.5037 - accuracy: 0.7900␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
4/13 [========>.....................] - ETA: 0s - loss: 0.5465 - accuracy: 0.7525␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
7/13 [===============>..............] - ETA: 0s - loss: 0.5472 - accuracy: 0.7543␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
10/13 [======================>.......] - ETA: 0s - loss: 0.5380 - accuracy: 0.7770
Epoch 9: val_loss did not improve from 0.65189
␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
13/13 [==============================] - 0s 24ms/step - loss: 0.5348 - accuracy: 0.7766 - val_loss: 0.6528 - val_accuracy: 0.6406
Epoch 10/10
1/13 [=>............................] - ETA: 0s - loss: 0.5006 - accuracy: 0.7800␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
4/13 [========>.....................] - ETA: 0s - loss: 0.5241 - accuracy: 0.7500␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
7/13 [===============>..............] - ETA: 0s - loss: 0.5183 - accuracy: 0.7714␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
10/13 [======================>.......] - ETA: 0s - loss: 0.5131 - accuracy: 0.7870␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
13/13 [==============================] - ETA: 0s - loss: 0.5108 - accuracy: 0.7906
Epoch 10: val_loss did not improve from 0.65189
␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈␈
13/13 [==============================] - 0s 22ms/step - loss: 0.5108 - accuracy: 0.7906 - val_loss: 0.7780 - val_accuracy: 0.5469
: Getting training history for item:0 name = 'loss'
: Getting training history for item:1 name = 'accuracy'
: Getting training history for item:2 name = 'val_loss'
: Getting training history for item:3 name = 'val_accuracy'
: Elapsed time for training with 1600 events: 6.06 sec
: Setting up tf.keras
: Using TensorFlow version 2
: Use Keras version from TensorFlow : tf.keras
: Applying GPU option: gpu_options.allow_growth=True
: Disabled TF eager execution when evaluating model
: Loading Keras Model
: Loaded model from file: trained_model_cnn.h5
PyKeras : [dataset] : Evaluation of PyKeras on training sample (1600 events)
: Elapsed time for evaluation of 1600 events: 0.29 sec
: Creating xml weight file: ␛[0;36mdataset/weights/TMVA_CNN_Classification_PyKeras.weights.xml␛[0m
: Creating standalone class: ␛[0;36mdataset/weights/TMVA_CNN_Classification_PyKeras.class.C␛[0m
Factory : Training finished
:
Factory : Train method: PyTorch for Classification
:
:
: ␛[1m================================================================␛[0m
: ␛[1mH e l p f o r M V A m e t h o d [ PyTorch ] :␛[0m
:
: PyTorch is a scientific computing package supporting
: automatic differentiation. This method wraps the training
: and predictions steps of the PyTorch Python package for
: TMVA, so that dataloading, preprocessing and evaluation
: can be done within the TMVA system. To use this PyTorch
: interface, you need to generatea model with PyTorch first.
: Then, this model can be loaded and trained in TMVA.
:
:
: <Suppress this message by specifying "!H" in the booking option>
: ␛[1m================================================================␛[0m
:
: Split TMVA training data in 1280 training events and 320 validation events
: Print Training Model Architecture
: Option SaveBestOnly: Only model weights with smallest validation loss will be stored
: Elapsed time for training with 1600 events: 23.3 sec
PyTorch : [dataset] : Evaluation of PyTorch on training sample (1600 events)
: Elapsed time for evaluation of 1600 events: 0.424 sec
: Creating xml weight file: ␛[0;36mdataset/weights/TMVA_CNN_Classification_PyTorch.weights.xml␛[0m
: Creating standalone class: ␛[0;36mdataset/weights/TMVA_CNN_Classification_PyTorch.class.C␛[0m
Factory : Training finished
:
: Ranking input variables (method specific)...
BDT : Ranking result (top variable is best ranked)
: --------------------------------------
: Rank : Variable : Variable Importance
: --------------------------------------
: 1 : vars : 1.179e-02
: 2 : vars : 1.030e-02
: 3 : vars : 9.587e-03
: 4 : vars : 8.906e-03
: 5 : vars : 8.623e-03
: 6 : vars : 8.398e-03
: 7 : vars : 8.307e-03
: 8 : vars : 8.229e-03
: 9 : vars : 7.947e-03
: 10 : vars : 7.840e-03
: 11 : vars : 7.784e-03
: 12 : vars : 7.756e-03
: 13 : vars : 7.698e-03
: 14 : vars : 7.628e-03
: 15 : vars : 7.609e-03
: 16 : vars : 7.591e-03
: 17 : vars : 7.510e-03
: 18 : vars : 7.497e-03
: 19 : vars : 7.304e-03
: 20 : vars : 7.199e-03
: 21 : vars : 7.134e-03
: 22 : vars : 7.122e-03
: 23 : vars : 7.116e-03
: 24 : vars : 7.035e-03
: 25 : vars : 7.034e-03
: 26 : vars : 6.997e-03
: 27 : vars : 6.840e-03
: 28 : vars : 6.790e-03
: 29 : vars : 6.790e-03
: 30 : vars : 6.779e-03
: 31 : vars : 6.762e-03
: 32 : vars : 6.745e-03
: 33 : vars : 6.716e-03
: 34 : vars : 6.697e-03
: 35 : vars : 6.640e-03
: 36 : vars : 6.573e-03
: 37 : vars : 6.484e-03
: 38 : vars : 6.468e-03
: 39 : vars : 6.451e-03
: 40 : vars : 6.448e-03
: 41 : vars : 6.363e-03
: 42 : vars : 6.222e-03
: 43 : vars : 6.188e-03
: 44 : vars : 6.169e-03
: 45 : vars : 6.155e-03
: 46 : vars : 6.033e-03
: 47 : vars : 6.025e-03
: 48 : vars : 6.015e-03
: 49 : vars : 5.997e-03
: 50 : vars : 5.992e-03
: 51 : vars : 5.972e-03
: 52 : vars : 5.970e-03
: 53 : vars : 5.964e-03
: 54 : vars : 5.899e-03
: 55 : vars : 5.788e-03
: 56 : vars : 5.732e-03
: 57 : vars : 5.701e-03
: 58 : vars : 5.699e-03
: 59 : vars : 5.669e-03
: 60 : vars : 5.661e-03
: 61 : vars : 5.630e-03
: 62 : vars : 5.625e-03
: 63 : vars : 5.602e-03
: 64 : vars : 5.598e-03
: 65 : vars : 5.572e-03
: 66 : vars : 5.520e-03
: 67 : vars : 5.506e-03
: 68 : vars : 5.488e-03
: 69 : vars : 5.485e-03
: 70 : vars : 5.450e-03
: 71 : vars : 5.419e-03
: 72 : vars : 5.402e-03
: 73 : vars : 5.397e-03
: 74 : vars : 5.365e-03
: 75 : vars : 5.357e-03
: 76 : vars : 5.350e-03
: 77 : vars : 5.324e-03
: 78 : vars : 5.283e-03
: 79 : vars : 5.275e-03
: 80 : vars : 5.271e-03
: 81 : vars : 5.229e-03
: 82 : vars : 5.142e-03
: 83 : vars : 5.110e-03
: 84 : vars : 5.091e-03
: 85 : vars : 5.087e-03
: 86 : vars : 5.078e-03
: 87 : vars : 5.054e-03
: 88 : vars : 5.034e-03
: 89 : vars : 5.034e-03
: 90 : vars : 5.012e-03
: 91 : vars : 4.952e-03
: 92 : vars : 4.948e-03
: 93 : vars : 4.944e-03
: 94 : vars : 4.900e-03
: 95 : vars : 4.895e-03
: 96 : vars : 4.890e-03
: 97 : vars : 4.889e-03
: 98 : vars : 4.829e-03
: 99 : vars : 4.817e-03
: 100 : vars : 4.800e-03
: 101 : vars : 4.774e-03
: 102 : vars : 4.692e-03
: 103 : vars : 4.684e-03
: 104 : vars : 4.683e-03
: 105 : vars : 4.627e-03
: 106 : vars : 4.479e-03
: 107 : vars : 4.474e-03
: 108 : vars : 4.473e-03
: 109 : vars : 4.409e-03
: 110 : vars : 4.409e-03
: 111 : vars : 4.406e-03
: 112 : vars : 4.401e-03
: 113 : vars : 4.363e-03
: 114 : vars : 4.340e-03
: 115 : vars : 4.330e-03
: 116 : vars : 4.311e-03
: 117 : vars : 4.231e-03
: 118 : vars : 4.185e-03
: 119 : vars : 4.108e-03
: 120 : vars : 4.106e-03
: 121 : vars : 4.088e-03
: 122 : vars : 4.080e-03
: 123 : vars : 4.036e-03
: 124 : vars : 4.022e-03
: 125 : vars : 4.000e-03
: 126 : vars : 3.959e-03
: 127 : vars : 3.929e-03
: 128 : vars : 3.892e-03
: 129 : vars : 3.808e-03
: 130 : vars : 3.799e-03
: 131 : vars : 3.780e-03
: 132 : vars : 3.774e-03
: 133 : vars : 3.665e-03
: 134 : vars : 3.630e-03
: 135 : vars : 3.629e-03
: 136 : vars : 3.607e-03
: 137 : vars : 3.578e-03
: 138 : vars : 3.565e-03
: 139 : vars : 3.536e-03
: 140 : vars : 3.510e-03
: 141 : vars : 3.491e-03
: 142 : vars : 3.394e-03
: 143 : vars : 3.382e-03
: 144 : vars : 3.356e-03
: 145 : vars : 3.353e-03
: 146 : vars : 3.306e-03
: 147 : vars : 3.306e-03
: 148 : vars : 3.283e-03
: 149 : vars : 3.280e-03
: 150 : vars : 3.257e-03
: 151 : vars : 3.234e-03
: 152 : vars : 3.227e-03
: 153 : vars : 3.223e-03
: 154 : vars : 3.183e-03
: 155 : vars : 3.178e-03
: 156 : vars : 3.175e-03
: 157 : vars : 3.158e-03
: 158 : vars : 3.143e-03
: 159 : vars : 3.142e-03
: 160 : vars : 3.135e-03
: 161 : vars : 3.133e-03
: 162 : vars : 3.113e-03
: 163 : vars : 3.101e-03
: 164 : vars : 3.098e-03
: 165 : vars : 3.092e-03
: 166 : vars : 3.088e-03
: 167 : vars : 3.058e-03
: 168 : vars : 3.049e-03
: 169 : vars : 3.029e-03
: 170 : vars : 3.029e-03
: 171 : vars : 3.015e-03
: 172 : vars : 2.993e-03
: 173 : vars : 2.964e-03
: 174 : vars : 2.946e-03
: 175 : vars : 2.922e-03
: 176 : vars : 2.899e-03
: 177 : vars : 2.896e-03
: 178 : vars : 2.877e-03
: 179 : vars : 2.874e-03
: 180 : vars : 2.873e-03
: 181 : vars : 2.861e-03
: 182 : vars : 2.858e-03
: 183 : vars : 2.822e-03
: 184 : vars : 2.806e-03
: 185 : vars : 2.767e-03
: 186 : vars : 2.755e-03
: 187 : vars : 2.746e-03
: 188 : vars : 2.717e-03
: 189 : vars : 2.702e-03
: 190 : vars : 2.666e-03
: 191 : vars : 2.658e-03
: 192 : vars : 2.622e-03
: 193 : vars : 2.606e-03
: 194 : vars : 2.604e-03
: 195 : vars : 2.573e-03
: 196 : vars : 2.543e-03
: 197 : vars : 2.514e-03
: 198 : vars : 2.467e-03
: 199 : vars : 2.446e-03
: 200 : vars : 2.334e-03
: 201 : vars : 2.327e-03
: 202 : vars : 2.266e-03
: 203 : vars : 2.251e-03
: 204 : vars : 2.193e-03
: 205 : vars : 2.186e-03
: 206 : vars : 2.161e-03
: 207 : vars : 2.156e-03
: 208 : vars : 1.821e-03
: 209 : vars : 1.738e-03
: 210 : vars : 1.731e-03
: 211 : vars : 1.462e-03
: 212 : vars : 1.401e-03
: 213 : vars : 1.221e-03
: 214 : vars : 1.146e-03
: 215 : vars : 9.470e-04
: 216 : vars : 8.349e-04
: 217 : vars : 4.707e-04
: 218 : vars : 0.000e+00
: 219 : vars : 0.000e+00
: 220 : vars : 0.000e+00
: 221 : vars : 0.000e+00
: 222 : vars : 0.000e+00
: 223 : vars : 0.000e+00
: 224 : vars : 0.000e+00
: 225 : vars : 0.000e+00
: 226 : vars : 0.000e+00
: 227 : vars : 0.000e+00
: 228 : vars : 0.000e+00
: 229 : vars : 0.000e+00
: 230 : vars : 0.000e+00
: 231 : vars : 0.000e+00
: 232 : vars : 0.000e+00
: 233 : vars : 0.000e+00
: 234 : vars : 0.000e+00
: 235 : vars : 0.000e+00
: 236 : vars : 0.000e+00
: 237 : vars : 0.000e+00
: 238 : vars : 0.000e+00
: 239 : vars : 0.000e+00
: 240 : vars : 0.000e+00
: 241 : vars : 0.000e+00
: 242 : vars : 0.000e+00
: 243 : vars : 0.000e+00
: 244 : vars : 0.000e+00
: 245 : vars : 0.000e+00
: 246 : vars : 0.000e+00
: 247 : vars : 0.000e+00
: 248 : vars : 0.000e+00
: 249 : vars : 0.000e+00
: 250 : vars : 0.000e+00
: 251 : vars : 0.000e+00
: 252 : vars : 0.000e+00
: 253 : vars : 0.000e+00
: 254 : vars : 0.000e+00
: 255 : vars : 0.000e+00
: 256 : vars : 0.000e+00
: --------------------------------------
: No variable ranking supplied by classifier: TMVA_DNN_CPU
: No variable ranking supplied by classifier: TMVA_CNN_CPU
: No variable ranking supplied by classifier: PyKeras
: No variable ranking supplied by classifier: PyTorch
TH1.Print Name = TrainingHistory_TMVA_DNN_CPU_trainingError, Entries= 0, Total sum= 4.94545
TH1.Print Name = TrainingHistory_TMVA_DNN_CPU_valError, Entries= 0, Total sum= 7.01578
TH1.Print Name = TrainingHistory_TMVA_CNN_CPU_trainingError, Entries= 0, Total sum= 6.05219
TH1.Print Name = TrainingHistory_TMVA_CNN_CPU_valError, Entries= 0, Total sum= 6.51958
TH1.Print Name = TrainingHistory_PyKeras_'accuracy', Entries= 0, Total sum= 6.62344
TH1.Print Name = TrainingHistory_PyKeras_'loss', Entries= 0, Total sum= 6.77425
TH1.Print Name = TrainingHistory_PyKeras_'val_accuracy', Entries= 0, Total sum= 5.925
TH1.Print Name = TrainingHistory_PyKeras_'val_loss', Entries= 0, Total sum= 7.89074
Factory : === Destroy and recreate all methods via weight files for testing ===
:
: Reading weight file: ␛[0;36mdataset/weights/TMVA_CNN_Classification_BDT.weights.xml␛[0m
: Reading weight file: ␛[0;36mdataset/weights/TMVA_CNN_Classification_TMVA_DNN_CPU.weights.xml␛[0m
: Reading weight file: ␛[0;36mdataset/weights/TMVA_CNN_Classification_TMVA_CNN_CPU.weights.xml␛[0m
: Reading weight file: ␛[0;36mdataset/weights/TMVA_CNN_Classification_PyKeras.weights.xml␛[0m
: Reading weight file: ␛[0;36mdataset/weights/TMVA_CNN_Classification_PyTorch.weights.xml␛[0m
Factory : ␛[1mTest all methods␛[0m
Factory : Test method: BDT for Classification performance
:
BDT : [dataset] : Evaluation of BDT on testing sample (400 events)
: Elapsed time for evaluation of 400 events: 0.00652 sec
Factory : Test method: TMVA_DNN_CPU for Classification performance
:
: Evaluate deep neural network on CPU using batches with size = 400
:
TMVA_DNN_CPU : [dataset] : Evaluation of TMVA_DNN_CPU on testing sample (400 events)
: Elapsed time for evaluation of 400 events: 0.0187 sec
Factory : Test method: TMVA_CNN_CPU for Classification performance
:
: Evaluate deep neural network on CPU using batches with size = 400
:
TMVA_CNN_CPU : [dataset] : Evaluation of TMVA_CNN_CPU on testing sample (400 events)
: Elapsed time for evaluation of 400 events: 0.147 sec
Factory : Test method: PyKeras for Classification performance
:
: Setting up tf.keras
: Using TensorFlow version 2
: Use Keras version from TensorFlow : tf.keras
: Applying GPU option: gpu_options.allow_growth=True
: Disabled TF eager execution when evaluating model
: Loading Keras Model
: Loaded model from file: trained_model_cnn.h5
PyKeras : [dataset] : Evaluation of PyKeras on testing sample (400 events)
: Elapsed time for evaluation of 400 events: 0.184 sec
Factory : Test method: PyTorch for Classification performance
:
: Setup PyTorch Model for training
: Executing user initialization code from /home/sftnight/build/workspace/root-makedoc-v628/rootspi/rdoc/src/v6-28-00-patches.build/tutorials/tmva/PyTorch_Generate_CNN_Model.py
RecursiveScriptModule(
original_name=Sequential
(0): RecursiveScriptModule(original_name=Reshape)
(1): RecursiveScriptModule(original_name=Conv2d)
(2): RecursiveScriptModule(original_name=ReLU)
(3): RecursiveScriptModule(original_name=BatchNorm2d)
(4): RecursiveScriptModule(original_name=Conv2d)
(5): RecursiveScriptModule(original_name=ReLU)
(6): RecursiveScriptModule(original_name=MaxPool2d)
(7): RecursiveScriptModule(original_name=Flatten)
(8): RecursiveScriptModule(original_name=Linear)
(9): RecursiveScriptModule(original_name=ReLU)
(10): RecursiveScriptModule(original_name=Linear)
(11): RecursiveScriptModule(original_name=Sigmoid)
)
[1, 4] train loss: 1.008
[1, 8] train loss: 0.719
[1, 12] train loss: 0.696
[1] val loss: 0.693
[2, 4] train loss: 0.699
[2, 8] train loss: 0.690
[2, 12] train loss: 0.695
[2] val loss: 0.692
[3, 4] train loss: 0.689
[3, 8] train loss: 0.691
[3, 12] train loss: 0.685
[3] val loss: 0.681
[4, 4] train loss: 0.678
[4, 8] train loss: 0.677
[4, 12] train loss: 0.669
[4] val loss: 0.676
[5, 4] train loss: 0.648
[5, 8] train loss: 0.634
[5, 12] train loss: 0.623
[5] val loss: 0.649
[6, 4] train loss: 0.637
[6, 8] train loss: 0.622
[6, 12] train loss: 0.555
[6] val loss: 0.631
[7, 4] train loss: 0.564
[7, 8] train loss: 0.560
[7, 12] train loss: 0.458
[7] val loss: 0.530
[8, 4] train loss: 0.495
[8, 8] train loss: 0.506
[8, 12] train loss: 0.504
[8] val loss: 0.530
[9, 4] train loss: 0.532
[9, 8] train loss: 0.451
[9, 12] train loss: 0.408
[9] val loss: 0.539
[10, 4] train loss: 0.451
[10, 8] train loss: 0.448
[10, 12] train loss: 0.391
[10] val loss: 0.528
Finished Training on 10 Epochs!
running Torch code defining the model....
The PyTorch CNN model is created and saved as PyTorchModelCNN.pt
: Loaded pytorch train function:
: Loaded pytorch optimizer:
: Loaded pytorch loss function:
: Loaded pytorch predict function:
: Loaded model from file: PyTorchTrainedModelCNN.pt
PyTorch : [dataset] : Evaluation of PyTorch on testing sample (400 events)
: Elapsed time for evaluation of 400 events: 0.12 sec
Factory : ␛[1mEvaluate all methods␛[0m
Factory : Evaluate classifier: BDT
:
BDT : [dataset] : Loop over test events and fill histograms with classifier response...
:
: Dataset[dataset] : variable plots are not produces ! The number of variables is 256 , it is larger than 200
Factory : Evaluate classifier: TMVA_DNN_CPU
:
TMVA_DNN_CPU : [dataset] : Loop over test events and fill histograms with classifier response...
:
: Evaluate deep neural network on CPU using batches with size = 1000
:
: Dataset[dataset] : variable plots are not produces ! The number of variables is 256 , it is larger than 200
Factory : Evaluate classifier: TMVA_CNN_CPU
:
TMVA_CNN_CPU : [dataset] : Loop over test events and fill histograms with classifier response...
:
: Evaluate deep neural network on CPU using batches with size = 1000
:
: Dataset[dataset] : variable plots are not produces ! The number of variables is 256 , it is larger than 200
Factory : Evaluate classifier: PyKeras
:
PyKeras : [dataset] : Loop over test events and fill histograms with classifier response...
:
: Dataset[dataset] : variable plots are not produces ! The number of variables is 256 , it is larger than 200
Factory : Evaluate classifier: PyTorch
:
PyTorch : [dataset] : Loop over test events and fill histograms with classifier response...
:
: Dataset[dataset] : variable plots are not produces ! The number of variables is 256 , it is larger than 200
:
: Evaluation results ranked by best signal efficiency and purity (area)
: -------------------------------------------------------------------------------------------------------------------
: DataSet MVA
: Name: Method: ROC-integ
: dataset PyTorch : 0.843
: dataset TMVA_CNN_CPU : 0.804
: dataset BDT : 0.704
: dataset PyKeras : 0.685
: dataset TMVA_DNN_CPU : 0.684
: -------------------------------------------------------------------------------------------------------------------
:
: Testing efficiency compared to training efficiency (overtraining check)
: -------------------------------------------------------------------------------------------------------------------
: DataSet MVA Signal efficiency: from test sample (from training sample)
: Name: Method: @B=0.01 @B=0.10 @B=0.30
: -------------------------------------------------------------------------------------------------------------------
: dataset PyTorch : 0.125 (0.335) 0.595 (0.635) 0.805 (0.831)
: dataset TMVA_CNN_CPU : 0.250 (0.295) 0.525 (0.645) 0.700 (0.870)
: dataset BDT : 0.055 (0.195) 0.320 (0.570) 0.613 (0.817)
: dataset PyKeras : 0.065 (0.055) 0.375 (0.397) 0.565 (0.620)
: dataset TMVA_DNN_CPU : 0.045 (0.085) 0.218 (0.495) 0.575 (0.800)
: -------------------------------------------------------------------------------------------------------------------
:
Dataset:dataset : Created tree 'TestTree' with 400 events
:
Dataset:dataset : Created tree 'TrainTree' with 1600 events
:
Factory : ␛[1mThank you for using TMVA!␛[0m
: ␛[1mFor citation information, please visit: http://tmva.sf.net/citeTMVA.html␛[0m
/***
# TMVA Classification Example Using a Convolutional Neural Network
**/
/// Helper function to create input images data
/// we create a signal and background 2D histograms from 2d gaussians
/// with a location (means in X and Y) different for each event
/// The difference between signal and background is in the gaussian width.
/// The width for the background gaussian is slightly larger than the signal width by few % values
///
///
void MakeImagesTree(int n, int nh, int nw)
{
// image size (nh x nw)
const int ntot = nh * nw;
const TString fileOutName = TString::Format("images_data_%dx%d.root", nh, nw);
TFile f(fileOutName, "RECREATE");
const int nRndmEvts = 10000; // number of events we use to fill each image
double delta_sigma = 0.1; // 5% difference in the sigma
double pixelNoise = 5;
double sX1 = 3;
double sY1 = 3;
double sX2 = sX1 + delta_sigma;
double sY2 = sY1 - delta_sigma;
TH2D h1("h1", "h1", nh, 0, 10, nw, 0, 10);
TH2D h2("h2", "h2", nh, 0, 10, nw, 0, 10);
TF2 f1("f1", "xygaus");
TF2 f2("f2", "xygaus");
TTree sgn("sig_tree", "signal_tree");
TTree bkg("bkg_tree", "background_tree");
std::vector<float> x1(ntot);
std::vector<float> x2(ntot);
// create signal and background trees with a single branch
// an std::vector<float> of size nh x nw containing the image data
std::vector<float> *px1 = &x1;
std::vector<float> *px2 = &x2;
bkg.Branch("vars", "std::vector<float>", &px1);
sgn.Branch("vars", "std::vector<float>", &px2);
// std::cout << "create tree " << std::endl;
sgn.SetDirectory(&f);
bkg.SetDirectory(&f);
f1.SetParameters(1, 5, sX1, 5, sY1);
f2.SetParameters(1, 5, sX2, 5, sY2);
gRandom->SetSeed(0);
std::cout << "Filling ROOT tree " << std::endl;
for (int i = 0; i < n; ++i) {
if (i % 1000 == 0)
std::cout << "Generating image event ... " << i << std::endl;
h1.Reset();
h2.Reset();
// generate random means in range [3,7] to be not too much on the border
f1.SetParameter(1, gRandom->Uniform(3, 7));
f1.SetParameter(3, gRandom->Uniform(3, 7));
f2.SetParameter(1, gRandom->Uniform(3, 7));
f2.SetParameter(3, gRandom->Uniform(3, 7));
h1.FillRandom("f1", nRndmEvts);
h2.FillRandom("f2", nRndmEvts);
for (int k = 0; k < nh; ++k) {
for (int l = 0; l < nw; ++l) {
int m = k * nw + l;
// add some noise in each bin
x1[m] = h1.GetBinContent(k + 1, l + 1) + gRandom->Gaus(0, pixelNoise);
x2[m] = h2.GetBinContent(k + 1, l + 1) + gRandom->Gaus(0, pixelNoise);
}
}
sgn.Fill();
bkg.Fill();
}
sgn.Write();
bkg.Write();
Info("MakeImagesTree", "Signal and background tree with images data written to the file %s", f.GetName());
sgn.Print();
bkg.Print();
f.Close();
}
/// @brief Run the TMVA CNN Classification example
/// @param nevts : number of signal/background events. Use by default a low value (1000)
/// but increase to at least 5000 to get a good result
/// @param opt : vector of bool with method used (default all on if available). The order is:
/// - TMVA CNN
/// - Keras CNN
/// - TMVA DNN
/// - TMVA BDT
/// - PyTorch CNN
void TMVA_CNN_Classification(int nevts = 1000, std::vector<bool> opt = {1, 1, 1, 1, 1})
{
int imgSize = 16 * 16;
TString inputFileName = "images_data_16x16.root";
bool fileExist = !gSystem->AccessPathName(inputFileName);
// if file does not exists create it
if (!fileExist) {
MakeImagesTree(nevts, 16, 16);
}
bool useTMVACNN = (opt.size() > 0) ? opt[0] : false;
bool useKerasCNN = (opt.size() > 1) ? opt[1] : false;
bool useTMVADNN = (opt.size() > 2) ? opt[2] : false;
bool useTMVABDT = (opt.size() > 3) ? opt[3] : false;
bool usePyTorchCNN = (opt.size() > 4) ? opt[4] : false;
#ifndef R__HAS_TMVACPU
#ifndef R__HAS_TMVAGPU
Warning("TMVA_CNN_Classification",
"TMVA is not build with GPU or CPU multi-thread support. Cannot use TMVA Deep Learning for CNN");
useTMVACNN = false;
#endif
#endif
bool writeOutputFile = true;
#ifdef R__USE_IMT
int num_threads = 4; // use by default 4 threads if value is not set before
// switch off MT in OpenBLAS to avoid conflict with tbb
gSystem->Setenv("OMP_NUM_THREADS", "1");
// do enable MT running
if (num_threads >= 0) {
ROOT::EnableImplicitMT(num_threads);
}
#endif
TMVA::Tools::Instance();
std::cout << "Running with nthreads = " << ROOT::GetThreadPoolSize() << std::endl;
#ifdef R__HAS_PYMVA
gSystem->Setenv("KERAS_BACKEND", "tensorflow");
// for using Keras
TMVA::PyMethodBase::PyInitialize();
#else
useKerasCNN = false;
usePyTorchCNN = false;
#endif
TFile *outputFile = nullptr;
if (writeOutputFile)
outputFile = TFile::Open("TMVA_CNN_ClassificationOutput.root", "RECREATE");
/***
## Create TMVA Factory
Create the Factory class. Later you can choose the methods
whose performance you'd like to investigate.
The factory is the major TMVA object you have to interact with. Here is the list of parameters you need to pass
- The first argument is the base of the name of all the output
weight files in the directory weight/ that will be created with the
method parameters
- The second argument is the output file for the training results
- The third argument is a string option defining some general configuration for the TMVA session.
For example all TMVA output can be suppressed by removing the "!" (not) in front of the "Silent" argument in the
option string
- note that we disable any pre-transformation of the input variables and we avoid computing correlations between
input variables
***/
TMVA::Factory factory(
"TMVA_CNN_Classification", outputFile,
"!V:ROC:!Silent:Color:AnalysisType=Classification:Transformations=None:!Correlations");
/***
## Declare DataLoader(s)
The next step is to declare the DataLoader class that deals with input variables
Define the input variables that shall be used for the MVA training
note that you may also use variable expressions, which can be parsed by TTree::Draw( "expression" )]
In this case the input data consists of an image of 16x16 pixels. Each single pixel is a branch in a ROOT TTree
**/
TMVA::DataLoader loader("dataset");
/***
## Setup Dataset(s)
Define input data file and signal and background trees
**/
std::unique_ptr<TFile> inputFile{TFile::Open(inputFileName)};
if (!inputFile) {
Error("TMVA_CNN_Classification", "Error opening input file %s - exit", inputFileName.Data());
return;
}
// --- Register the training and test trees
auto signalTree = inputFile->Get<TTree>("sig_tree");
auto backgroundTree = inputFile->Get<TTree>("bkg_tree");
if (!signalTree) {
Error("TMVA_CNN_Classification", "Could not find signal tree in file '%s'", inputFileName.Data());
return;
}
if (!backgroundTree) {
Error("TMVA_CNN_Classification", "Could not find background tree in file '%s'", inputFileName.Data());
return;
}
int nEventsSig = signalTree->GetEntries();
int nEventsBkg = backgroundTree->GetEntries();
// global event weights per tree (see below for setting event-wise weights)
Double_t signalWeight = 1.0;
Double_t backgroundWeight = 1.0;
// You can add an arbitrary number of signal or background trees
loader.AddSignalTree(signalTree, signalWeight);
loader.AddBackgroundTree(backgroundTree, backgroundWeight);
/// add event variables (image)
/// use new method (from ROOT 6.20 to add a variable array for all image data)
loader.AddVariablesArray("vars", imgSize);
// Set individual event weights (the variables must exist in the original TTree)
// for signal : factory->SetSignalWeightExpression ("weight1*weight2");
// for background: factory->SetBackgroundWeightExpression("weight1*weight2");
// loader.SetBackgroundWeightExpression( "weight" );
// Apply additional cuts on the signal and background samples (can be different)
TCut mycuts = ""; // for example: TCut mycuts = "abs(var1)<0.5 && abs(var2-0.5)<1";
TCut mycutb = ""; // for example: TCut mycutb = "abs(var1)<0.5";
// Tell the factory how to use the training and testing events
//
// If no numbers of events are given, half of the events in the tree are used
// for training, and the other half for testing:
// loader.PrepareTrainingAndTestTree( mycut, "SplitMode=random:!V" );
// It is possible also to specify the number of training and testing events,
// note we disable the computation of the correlation matrix of the input variables
int nTrainSig = 0.8 * nEventsSig;
int nTrainBkg = 0.8 * nEventsBkg;
// build the string options for DataLoader::PrepareTrainingAndTestTree
TString prepareOptions = TString::Format(
"nTrain_Signal=%d:nTrain_Background=%d:SplitMode=Random:SplitSeed=100:NormMode=NumEvents:!V:!CalcCorrelations",
nTrainSig, nTrainBkg);
loader.PrepareTrainingAndTestTree(mycuts, mycutb, prepareOptions);
/***
DataSetInfo : [dataset] : Added class "Signal"
: Add Tree sig_tree of type Signal with 10000 events
DataSetInfo : [dataset] : Added class "Background"
: Add Tree bkg_tree of type Background with 10000 events
**/
/****
# Booking Methods
Here we book the TMVA methods. We book a Boosted Decision Tree method (BDT)
**/
// Boosted Decision Trees
if (useTMVABDT) {
factory.BookMethod(&loader, TMVA::Types::kBDT, "BDT",
"!V:NTrees=200:MinNodeSize=2.5%:MaxDepth=2:BoostType=AdaBoost:AdaBoostBeta=0.5:"
"UseBaggedBoost:BaggedSampleFraction=0.5:SeparationType=GiniIndex:nCuts=20");
}
/**
#### Booking Deep Neural Network
Here we book the DNN of TMVA. See the example TMVA_Higgs_Classification.C for a detailed description of the
options
**/
if (useTMVADNN) {
TString layoutString(
"Layout=DENSE|100|RELU,BNORM,DENSE|100|RELU,BNORM,DENSE|100|RELU,BNORM,DENSE|100|RELU,DENSE|1|LINEAR");
// Training strategies
// one can catenate several training strings with different parameters (e.g. learning rates or regularizations
// parameters) The training string must be concatenates with the `|` delimiter
TString trainingString1("LearningRate=1e-3,Momentum=0.9,Repetitions=1,"
"ConvergenceSteps=5,BatchSize=100,TestRepetitions=1,"
"MaxEpochs=10,WeightDecay=1e-4,Regularization=None,"
"Optimizer=ADAM,DropConfig=0.0+0.0+0.0+0.");
TString trainingStrategyString("TrainingStrategy=");
trainingStrategyString += trainingString1; // + "|" + trainingString2 + ....
// Build now the full DNN Option string
TString dnnOptions("!H:V:ErrorStrategy=CROSSENTROPY:VarTransform=None:"
"WeightInitialization=XAVIER");
dnnOptions.Append(":");
dnnOptions.Append(layoutString);
dnnOptions.Append(":");
dnnOptions.Append(trainingStrategyString);
TString dnnMethodName = "TMVA_DNN_CPU";
// use GPU if available
#ifdef R__HAS_TMVAGPU
dnnOptions += ":Architecture=GPU";
dnnMethodName = "TMVA_DNN_GPU";
#elif defined(R__HAS_TMVACPU)
dnnOptions += ":Architecture=CPU";
#endif
factory.BookMethod(&loader, TMVA::Types::kDL, dnnMethodName, dnnOptions);
}
/***
### Book Convolutional Neural Network in TMVA
For building a CNN one needs to define
- Input Layout : number of channels (in this case = 1) | image height | image width
- Batch Layout : batch size | number of channels | image size = (height*width)
Then one add Convolutional layers and MaxPool layers.
- For Convolutional layer the option string has to be:
- CONV | number of units | filter height | filter width | stride height | stride width | padding height | paddig
width | activation function
- note in this case we are using a filer 3x3 and padding=1 and stride=1 so we get the output dimension of the
conv layer equal to the input
- note we use after the first convolutional layer a batch normalization layer. This seems to help significantly the
convergence
- For the MaxPool layer:
- MAXPOOL | pool height | pool width | stride height | stride width
The RESHAPE layer is needed to flatten the output before the Dense layer
Note that to run the CNN is required to have CPU or GPU support
***/
if (useTMVACNN) {
TString inputLayoutString("InputLayout=1|16|16");
// Batch Layout
TString layoutString("Layout=CONV|10|3|3|1|1|1|1|RELU,BNORM,CONV|10|3|3|1|1|1|1|RELU,MAXPOOL|2|2|1|1,"
"RESHAPE|FLAT,DENSE|100|RELU,DENSE|1|LINEAR");
// Training strategies.
TString trainingString1("LearningRate=1e-3,Momentum=0.9,Repetitions=1,"
"ConvergenceSteps=5,BatchSize=100,TestRepetitions=1,"
"MaxEpochs=10,WeightDecay=1e-4,Regularization=None,"
"Optimizer=ADAM,DropConfig=0.0+0.0+0.0+0.0");
TString trainingStrategyString("TrainingStrategy=");
trainingStrategyString +=
trainingString1; // + "|" + trainingString2 + "|" + trainingString3; for concatenating more training strings
// Build full CNN Options.
TString cnnOptions("!H:V:ErrorStrategy=CROSSENTROPY:VarTransform=None:"
"WeightInitialization=XAVIER");
cnnOptions.Append(":");
cnnOptions.Append(inputLayoutString);
cnnOptions.Append(":");
cnnOptions.Append(layoutString);
cnnOptions.Append(":");
cnnOptions.Append(trainingStrategyString);
//// New DL (CNN)
TString cnnMethodName = "TMVA_CNN_CPU";
// use GPU if available
#ifdef R__HAS_TMVAGPU
cnnOptions += ":Architecture=GPU";
cnnMethodName = "TMVA_CNN_GPU";
#else
cnnOptions += ":Architecture=CPU";
cnnMethodName = "TMVA_CNN_CPU";
#endif
factory.BookMethod(&loader, TMVA::Types::kDL, cnnMethodName, cnnOptions);
}
/**
### Book Convolutional Neural Network in Keras using a generated model
**/
#ifdef R__HAS_PYMVA
// The next section uses Python packages, execute it only if PyMVA is available
TString tmva_python_exe{TMVA::Python_Executable()};
TString python_exe = tmva_python_exe.IsNull() ? "python" : tmva_python_exe;
if (useKerasCNN) {
Info("TMVA_CNN_Classification", "Building convolutional keras model");
// create python script which can be executed
// create 2 conv2d layer + maxpool + dense
TMacro m;
m.AddLine("import tensorflow");
m.AddLine("from tensorflow.keras.models import Sequential");
m.AddLine("from tensorflow.keras.optimizers import Adam");
m.AddLine(
"from tensorflow.keras.layers import Input, Dense, Dropout, Flatten, Conv2D, MaxPooling2D, Reshape, BatchNormalization");
m.AddLine("");
m.AddLine("model = Sequential() ");
m.AddLine("model.add(Reshape((16, 16, 1), input_shape = (256, )))");
m.AddLine("model.add(Conv2D(10, kernel_size = (3, 3), kernel_initializer = 'glorot_normal',activation = "
"'relu', padding = 'same'))");
m.AddLine("model.add(BatchNormalization())");
m.AddLine("model.add(Conv2D(10, kernel_size = (3, 3), kernel_initializer = 'glorot_normal',activation = "
"'relu', padding = 'same'))");
// m.AddLine("model.add(BatchNormalization())");
m.AddLine("model.add(MaxPooling2D(pool_size = (2, 2), strides = (1,1))) ");
m.AddLine("model.add(Flatten())");
m.AddLine("model.add(Dense(256, activation = 'relu')) ");
m.AddLine("model.add(Dense(2, activation = 'sigmoid')) ");
m.AddLine("model.compile(loss = 'binary_crossentropy', optimizer = Adam(learning_rate = 0.001), weighted_metrics = ['accuracy'])");
m.AddLine("model.save('model_cnn.h5')");
m.AddLine("model.summary()");
m.SaveSource("make_cnn_model.py");
// execute
gSystem->Exec(python_exe + " make_cnn_model.py");
if (gSystem->AccessPathName("model_cnn.h5")) {
Warning("TMVA_CNN_Classification", "Error creating Keras model file - skip using Keras");
} else {
// book PyKeras method only if Keras model could be created
Info("TMVA_CNN_Classification", "Booking tf.Keras CNN model");
factory.BookMethod(
&loader, TMVA::Types::kPyKeras, "PyKeras",
"H:!V:VarTransform=None:FilenameModel=model_cnn.h5:tf.keras:"
"FilenameTrainedModel=trained_model_cnn.h5:NumEpochs=10:BatchSize=100:"
"GpuOptions=allow_growth=True"); // needed for RTX NVidia card and to avoid TF allocates all GPU memory
}
}
if (usePyTorchCNN) {
Info("TMVA_CNN_Classification", "Using Convolutional PyTorch Model");
TString pyTorchFileName = gROOT->GetTutorialDir() + TString("/tmva/PyTorch_Generate_CNN_Model.py");
// check that pytorch can be imported and file defining the model and used later when booking the method is
// existing
if (gSystem->Exec(python_exe + " -c 'import torch'") || gSystem->AccessPathName(pyTorchFileName)) {
Warning("TMVA_CNN_Classification", "PyTorch is not installed or model building file is not existing - skip using PyTorch");
} else {
// book PyTorch method only if PyTorch model could be created
Info("TMVA_CNN_Classification", "Booking PyTorch CNN model");
TString methodOpt = "H:!V:VarTransform=None:FilenameModel=PyTorchModelCNN.pt:"
"FilenameTrainedModel=PyTorchTrainedModelCNN.pt:NumEpochs=10:BatchSize=100";
methodOpt += TString(":UserCode=") + pyTorchFileName;
factory.BookMethod(&loader, TMVA::Types::kPyTorch, "PyTorch", methodOpt);
}
}
#endif
//// ## Train Methods
factory.TrainAllMethods();
/// ## Test and Evaluate Methods
factory.TestAllMethods();
factory.EvaluateAllMethods();
/// ## Plot ROC Curve
auto c1 = factory.GetROCCurve(&loader);
c1->Draw();
// close outputfile to save output file
outputFile->Close();
}
f
#define f(i)
Definition RSha256.hxx:104
Double_t
double Double_t
Definition RtypesCore.h:59
Info
void Info(const char *location, const char *msgfmt,...)
Use this function for informational messages.
Definition TError.cxx:230
Error
void Error(const char *location, const char *msgfmt,...)
Use this function in case an error occurred.
Definition TError.cxx:197
Warning
void Warning(const char *location, const char *msgfmt,...)
Use this function in warning situations.
Definition TError.cxx:241
x2
Option_t Option_t TPoint TPoint const char x2
Definition TGWin32VirtualXProxy.cxx:70
x1
Option_t Option_t TPoint TPoint const char x1
Definition TGWin32VirtualXProxy.cxx:70
gROOT
#define gROOT
Definition TROOT.h:405
gRandom
R__EXTERN TRandom * gRandom
Definition TRandom.h:62
gSystem
R__EXTERN TSystem * gSystem
Definition TSystem.h:560
TCut
A specialized string object used for TTree selections.
Definition TCut.h:25
TF1::SetParameters
virtual void SetParameters(const Double_t *params)
Definition TF1.h:649
TF1::SetParameter
virtual void SetParameter(Int_t param, Double_t value)
Definition TF1.h:639
TF2
A 2-Dim function with parameters.
Definition TF2.h:29
TFile
A ROOT file is a suite of consecutive data records (TKey instances) with a well defined format.
Definition TFile.h:51
TFile::Open
static TFile * Open(const char *name, Option_t *option="", const char *ftitle="", Int_t compress=ROOT::RCompressionSetting::EDefaults::kUseCompiledDefault, Int_t netopt=0)
Create / open a file.
Definition TFile.cxx:4053
TFile::Close
void Close(Option_t *option="") override
Close a file.
Definition TFile.cxx:914
TH1F::Reset
void Reset(Option_t *option="") override
Reset.
Definition TH1.cxx:9979
TH1::FillRandom
virtual void FillRandom(const char *fname, Int_t ntimes=5000, TRandom *rng=nullptr)
Fill histogram following distribution in function fname.
Definition TH1.cxx:3513
TH1::GetBinContent
virtual Double_t GetBinContent(Int_t bin) const
Return content of bin number bin.
Definition TH1.cxx:5025
TH2D
2-D histogram with a double per channel (see TH1 documentation)}
Definition TH2.h:300
TMVA::Factory
This is the main MVA steering class.
Definition Factory.h:80
TMVA::PyMethodBase::PyInitialize
static void PyInitialize()
Initialize Python interpreter.
Definition PyMethodBase.cxx:153
TMVA::Tools::Instance
static Tools & Instance()
Definition Tools.cxx:71
TMacro
Class supporting a collection of lines with C++ code.
Definition TMacro.h:31
TRandom::Gaus
virtual Double_t Gaus(Double_t mean=0, Double_t sigma=1)
Samples a random number from the standard Normal (Gaussian) Distribution with the given mean and sigm...
Definition TRandom.cxx:274
TRandom::SetSeed
virtual void SetSeed(ULong_t seed=0)
Set the random generator seed.
Definition TRandom.cxx:608
TRandom::Uniform
virtual Double_t Uniform(Double_t x1=1)
Returns a uniform deviate on the interval (0, x1).
Definition TRandom.cxx:672
TString
Basic string class.
Definition TString.h:139
TString::Data
const char * Data() const
Definition TString.h:380
TString::IsNull
Bool_t IsNull() const
Definition TString.h:418
TString::Format
static TString Format(const char *fmt,...)
Static method which formats a string using a printf style format descriptor and return a TString.
Definition TString.cxx:2356
TSystem::Exec
virtual Int_t Exec(const char *shellcmd)
Execute a command.
Definition TSystem.cxx:656
TSystem::AccessPathName
virtual Bool_t AccessPathName(const char *path, EAccessMode mode=kFileExists)
Returns FALSE if one can access a file using the specified access mode.
Definition TSystem.cxx:1299
TSystem::Setenv
virtual void Setenv(const char *name, const char *value)
Set environment variable.
Definition TSystem.cxx:1650
TTree
A TTree represents a columnar dataset.
Definition TTree.h:79
TTree::GetEntries
virtual Long64_t GetEntries() const
Definition TTree.h:460
c1
return c1
Definition legend1.C:41
n
const Int_t n
Definition legend1.C:16
h1
TH1F * h1
Definition legend1.C:5
f1
TF1 * f1
Definition legend1.C:11
ROOT::EnableImplicitMT
void EnableImplicitMT(UInt_t numthreads=0)
Enable ROOT's implicit multi-threading for all objects and methods that provide an internal paralleli...
Definition TROOT.cxx:527
ROOT::GetThreadPoolSize
UInt_t GetThreadPoolSize()
Returns the size of ROOT's thread pool.
Definition TROOT.cxx:565
TMVA_CNN_Classification
Definition TMVA_CNN_Classification.py:1
TMVA::Python_Executable
TString Python_Executable()
Function to find current Python executable used by ROOT If Python2 is installed return "python" Inste...
Definition PyMethodBase.cxx:43
m
TMarker m
Definition textangle.C:8
l
TLine l
Definition textangle.C:4
Author
Lorenzo Moneta

Definition in file TMVA_CNN_Classification.C.