ReconstructionModel = Model2DViewsTo3D(
Name,
View1,
View2,
Width,
Depth,
BatchSize,
NClasses,
init=TestDefaultParam("WeightInitialization", 'normal'),
#activation=TestDefaultParam("activation","relu"),
Dropout=TestDefaultParam("DropoutLayers", 0.5),
BatchNormalization=TestDefaultParam("BatchNormLayers", False),
OutputBase=OutputBase)
ReconstructionModel.Build()
print " Done."
print "Output Directory:", ReconstructionModel.OutDir
# Store the Configuration Dictionary
ReconstructionModel.MetaData["Configuration"] = Config
if "HyperParamSet" in dir():
ReconstructionModel.MetaData["HyperParamSet"] = HyperParamSet
# Print out the Model Summary
ReconstructionModel.Model.summary()
# Compile The Model
print "Compiling Model."
ReconstructionModel.BuildOptimizer(optimizer, Config)
ReconstructionModel.Compile(Metrics=["accuracy"])
# BatchNormalization=TestDefaultParam("BatchNormLayers",False),
# NoClassificationLayer=ECAL and HCAL,
# OutputBase=OutputBase)
# HCALModel.Build()
# MyModel=HCALModel
if HCAL and ECAL:
MyModel = MergerModel(Name + "_Merged", [ECALModel, HCALModel],
NClasses,
WeightInitialization,
OutputBase=OutputBase)
#"""
# Configure the Optimizer, using optimizer configuration parameter.
MyModel.Loss = loss
# Build it
MyModel.Build()
print " Done."
if BuildModel:
print "Output Directory:", MyModel.OutDir
# Store the Configuration Dictionary
MyModel.MetaData["Configuration"] = Config
if "HyperParamSet" in dir():
MyModel.MetaData["HyperParamSet"] = HyperParamSet
# Print out the Model Summary
MyModel.Model.summary()
# Compile The Model
print "Compiling Model."
MyModel.BuildOptimizer(optimizer, Config)
InDir=os.path.dirname(LoadModel),
OutputBase=OutputBase)
MyModel.Load(LoadModel)
if BuildModel and not MyModel.Model:
FailedLoad = True
else:
FailedLoad = False
# Or Build the model from scratch
if BuildModel and not MyModel.Model:
import keras
print "Building Model...",
MyModel = ConfigModel
MyModel.OutDir = OutputBase # adds .Test suffix when necessary
MyModel.Build() # Build it
print " Done."
if BuildModel:
print "Output Directory:", MyModel.OutDir
# Store the Configuration Dictionary
MyModel.MetaData["Configuration"] = Config
if "HyperParamSet" in dir():
MyModel.MetaData["HyperParamSet"] = HyperParamSet
# Print out the Model Summary
MyModel.Model.summary()
# Compile The Model
print "Compiling Model."
MyModel.BuildOptimizer(optimizer, Config)