Is ROOT TMVA supporting Recurrent/Convolution Neural Network?

Hello all,

Is ROOT TMVA supporting Recurrent/LSTM/Convolution Neural Network?

Thank you,

I think @moneta or @sitong should know …


We have currently a vanilla implementation for RNN, but not yet working on GPU, and we are including now LSTM and GRU. They will be available in the master soon, we have already a PR we are reviewing, (see

Convolutional networks are already supported both on CPU and GPU.
An example for convolution rural network is this notebook:


1 Like

Thank you very much for your reply,

I just curious how to you prepare the data for the RNN, it should be 3D tensor (batch_size, time_steps, features) and TMVA data-loader accepts TTrees.

Thank you in advance,