Hi An Tran,
The metrics returned by the inspector’s training logs are hardcoded 3. The TF-DF API does not currently allow to customize it.
If of importance, please file a feature request.
In the mean time, a (relatively inefficient) solution is to remove trees from the model (using the model inspector 2 and model builder 1) and then to evaluate each intermediate model with metrics=[tf.keras.metrics.AUC()].