How to save the name of a feature when saving model feature importance

[hzy0110]

hello
I use the following code

 with R.start(experiment_name=EXP_NAME):
    mlflow.lightgbm.autolog()
    R.log_params(**flatten_dict(model._params)) 
    model.fit(dataset=ds) 

Feature importance is saved
But the saved name is Column_XX
What I want saved is the factor name, such as ROC5

I know it can be obtained via
ds.handler.get_feature_config()
But I want to automatically record the original name.