doc/master/RBatchGenerator__TensorFlow_8py.html

import tensorflow as tf

import ROOT


tree_name = "sig_tree"

file_name = "http://root.cern/files/Higgs_data.root"


batch_size = 128

chunk_size = 5_000


target = "Type"


# Returns two TF.Dataset for training and validation batches.

ds_train, ds_valid = ROOT.TMVA.Experimental.CreateTFDatasets(

    tree_name,

    file_name,

    batch_size,

    chunk_size,

    validation_split=0.3,

    target=target,

)


# Get a list of the columns used for training

input_columns = ds_train.train_columns

num_features = len(input_columns)


##############################################################################

# AI example

##############################################################################


# Define TensorFlow model

model = tf.keras.Sequential(

    [

        tf.keras.layers.Dense(

            300, activation=tf.nn.tanh, input_shape=(num_features,)

        ),  # input shape required

        tf.keras.layers.Dense(300, activation=tf.nn.tanh),

        tf.keras.layers.Dense(300, activation=tf.nn.tanh),

        tf.keras.layers.Dense(1, activation=tf.nn.sigmoid),

    ]

)

loss_fn = tf.keras.losses.BinaryCrossentropy()

model.compile(optimizer="adam", loss=loss_fn, metrics=["accuracy"])


# Train model

model.fit(ds_train, validation_data=ds_valid, epochs=2)

len
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void char Point_t Rectangle_t WindowAttributes_t Float_t Float_t Float_t Int_t Int_t UInt_t UInt_t Rectangle_t Int_t Int_t Window_t TString Int_t GCValues_t GetPrimarySelectionOwner GetDisplay GetScreen GetColormap GetNativeEvent const char const char dpyName wid window const char font_name cursor keysym reg const char only_if_exist regb h Point_t winding char text const char depth char const char Int_t count const char ColorStruct_t color const char Pixmap_t Pixmap_t PictureAttributes_t attr const char char ret_data h unsigned char height h Atom_t Int_t ULong_t ULong_t unsigned char prop_list Atom_t Atom_t Atom_t Time_t UChar_t len
Definition TGWin32VirtualXProxy.cxx:249