def _compile_model(
self,
loss_function: Callable,
ground_distance_path: Path,
**loss_function_kwargs
):
if loss_function == self_guided_earth_mover_distance:
self.emd_weight_head_start = EmdWeightHeadStart()
self.ground_distance_manager = GroundDistanceManager(ground_distance_path)
lr_schedule = ExponentialDecay(
self.learning_rate,
decay_steps=429,
decay_rate=0.995
)
self.compile(
loss=loss_function(
model=self,
**loss_function_kwargs
),
optimizer=self._OPTIMIZER(
learning_rate=lr_schedule,
nesterov=True,
momentum=self._OPTIMIZER_MOMENTUM
),
metrics=self._METRICS,
run_eagerly=True
)
def alxs(lr_index, fold_index):
ground_distance_manager = GroundDistanceManager(
Path('ground_distances'))
ground_distance_matrix = ground_distance_manager.load_ground_distance_matrix(
'159')
evaluate_adience_model(evaluation_model=Alxs,
learning_rate_index=lr_index,
fold_index=fold_index,
loss_function=approximate_earth_mover_distance,
final_activation=softmax,
entropic_regularizer=10,
distance_matrix=ground_distance_matrix,
matrix_scaling_operations=100)