#cb += [plotClusteringDuringTraining(
# use_backgather_idx=8 + i,
# outputfile=train.outputDir + "/localclust/cluster_" + str(i) + '_',
# samplefile=samplepath,
# after_n_batches=500,
# on_epoch_end=False,
# publish=None,
# use_event=0)
# for i in [0, 2, 4]]
#
cb += [
plotEventDuringTraining(
outputfile=train.outputDir + "/condensation/c_"+str(i),
samplefile=samplepath,
after_n_batches=2*plotfrequency,
batchsize=200000,
on_epoch_end=False,
publish=None,
use_event=i)
for i in range(5)
]
from DeepJetCore.training.DeepJet_callbacks import simpleMetricsCallback
cb += [
simpleMetricsCallback(
output_file=train.outputDir+'/metrics.html',
record_frequency= 2,
plot_frequency = plotfrequency,
cb += [
plotClusteringDuringTraining(use_backgather_idx=8 + i,
outputfile=train.outputDir + "/plts/sn" +
str(i) + '_',
samplefile=samplepath,
after_n_batches=20,
on_epoch_end=False,
publish=None,
use_event=0) for i in [0, 2]
]
cb += [
plotEventDuringTraining(outputfile=train.outputDir + "/plts2/sn0",
samplefile=samplepath,
after_n_batches=20,
batchsize=200000,
on_epoch_end=False,
publish=None,
use_event=0)
]
cb += [
plotGravNetCoordsDuringTraining(
outputfile=train.outputDir + "/coords_" + str(i) + "/coord_" + str(i),
samplefile=samplepath,
after_n_batches=20,
batchsize=200000,
on_epoch_end=False,
publish=None,
use_event=0,
use_prediction_idx=i,