If I want to use your method on other datasets, how can I generate the tfrecord file ? Can you share the code for generating tfrecord?