Simple implementaion using custom dataset

Hi, 
Thank you for publishing the source codes! I am trying to train a WPFS model using my private dataset, but I am not sure how to build the model and training procedure. Can you kindly provide a simple implementation for a custom data set? 
Thank you very much!