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,
cb = [
plotClusteringDuringTraining(use_backgather_idx=7 + i,
outputfile=train.outputDir + "/plts/sn" +
str(i) + '_',
samplefile=samplepath,
after_n_batches=plot_after_batches,
on_epoch_end=False,
publish=publishpath + "_cl_" + str(i),
use_event=0) for i in [4, 5]
]
cb += [
plotEventDuringTraining(outputfile=train.outputDir + "/plts2/sn0",
samplefile=samplepath,
after_n_batches=plot_after_batches,
batchsize=200000,
on_epoch_end=False,
publish=publishpath + "_event_" + str(0),
use_event=0)
]
cb += [
plotGravNetCoordsDuringTraining(
outputfile=train.outputDir + "/coords_" + str(i) + "/coord_" + str(i),
samplefile=samplepath,
after_n_batches=plot_after_batches,
batchsize=200000,
on_epoch_end=False,
publish=publishpath + "_event_" + str(0),
use_event=0,
use_prediction_idx=i,