ModuleNotFoundError : but code from repo doesn't have this import-CodePudding

I'm trying to use the code from this Github repo : https://github.com/dairui01/TSU_evaluation/tree/main/Frame_map

When I replace the csv file there and the pkl file with my own, I get the following error in my terminal:

 (this is what I entered)python Frame_based_map.py -split CS -pkl_path ./data.pkl
 THis is the printout:
Traceback (most recent call last):
  File "Frame_based_map.py", line 47, in <module>
    logits = pickle.load(pkl, encoding='latin1')
ModuleNotFoundError: No module named 'models'

https://drive.google.com/drive/folders/1Y-EOwSNUT1r5dxTi1MP0668weZlnLfJ9?usp=sharing

This is the google drive link to the files I am using, with the github code. The files are generated using the same method as the github link, so I am not sure what is wrong with my data.pkl file.

Any help is appreciated :)

Thank you!

CodePudding user response：

data.pkl itself references the models module. The process of unpickling tries to load it and fails, since you do not have it.