Пользовательская целевая функция KerasTuner

Question

Я пытаюсь использовать Keras Tuner для точной настройки гиперпараметров. Я хотел бы максимизировать auc. Может ли кто-нибудь помочь мне с использованием kerastuner.Objective для пользовательского метри c?

EXECUTIONS_PER_TRIAL = 5

b_tuner = BayesianOptimization(
    tune_nn_model,
    objective='val_binary_accuracy',
    max_trials=MAX_TRIALS,
    executions_per_trial=EXECUTIONS_PER_TRIAL,
    directory='test_dir101897',
    project_name='b_tune_nn',
    seed=12347
)

Я попытался определить пользовательскую функцию, такую ​​как:

from sklearn import metrics
from keras import backend as K

def auc(y_true, y_pred):
    auc = tf.metrics.auc(y_true, y_pred)[1]
    K.get_session().run(tf.local_variables_initializer())
    return auc

и подключить ее к

objective='val_auc'

Но это не работает