change a python code

I want to someone make the below code as a function and run it on Oxford5k and Paris6k datasets.

https://github.com/ZER-0-NE/Image-Retrieval-using-Autoencoders
Tensorflow>2.4

Then add mAP and precision-recall curve and confusion matrix.
mAP

precison-recall based on
https://scikit-learn.org/stable/modules/generated/sklearn.metrics.precision_recall_curve.html

mAP based on
https://sniafas.medium.com/evaluating-information-retrieval-3975348ef97a

The function should accept a folder as train set and then the model should be trained on it.
Then the function accepts a folder as the test set with arbitrary number of images as query images.
For each image it must calculate mAP and precision-recall.

Confusion matrix or heatmap:
https://www.vlfeat.org/benchmarks/overview/retrieval.html