search

yashsmehta / personality-prediction

Experiments for automated personality detection using Language Models and psycholinguistic features on various famous personality datasets including the Essays dataset (Big-Five)
MIT License
187 stars 51 forks source link

finetuneNet.py #1

Closed sugarcane29 closed 4 years ago

sugarcane29 commented 4 years ago

Hi...I am unable to find the file "finetuneNet.py".

Could you please tell me where it is.

Thanks.

saminfatehi commented 4 years ago

Hi. After storing the embeddings in a pickle file (by running LM_extractor.py), you can use files in the "finetune_models" folder (as finetuneNet), for example, "MLP_LM.py".

sugarcane29 commented 4 years ago

okay. Thanks for the clarification. There are some inconsistencies with the .py files in the finetune_models folder. Some look for the .pkl file in the root folder, while some in the same folder. Case in point: LR.py Line 36 file = open('../' + inp_dir + dataset_type + '-' + embed + '-' + embed_mode + '-' + mode + '.pkl', 'rb') MLP_LM.py Line 46 file = open(inp_dir + dataset + '-' + embed + '-' + embed_mode + '-' + mode + '.pkl', 'rb')

Would be great if you could look into them.

Thanks.