target=target_train_TT,
masses=masses_train_TT,
weight=weight_train_TT,
label_model=args.label,
label_target='TT',
model_number=i)
instance_TT.BuildModel()
instance_TT.PlotHistory()
instance_TT.PrintModel()
print('-' * 80)
print('\n[INFO] Evaluating model')
print('\t[INFO] Valid weights')
mse_TT = instance_TT.UseModel(data=set_test_TT,
target=target_test_TT,
masses=masses_test_TT,
other=other_test_TT,
label_output='')
print('\t[INFO] Invalid weights')
mse_TT_inv = instance_TT.UseModel(data=set_invalid_TT,
target=np.zeros(set_invalid_TT.shape[0]),
masses=masses_invalid_TT,
other=other_invalid_TT,
label_output='invalid')
K.clear_session()
# Neural Net for Drell Yann weights #
print('=' * 80)
print("[INFO] Starting learning on DY weights")
for i in range(0, k):
print('\n[INFO] Cross validation step : ', i)
