eval_model.sh

CUDA_VISIBLE_DEVICES=3 python main.py --data_dir "./graph_data/ratebeer/medium_500_positive/" --graph_dir "./graph_data/ratebeer/medium_500_positive/geo_graph_batch/" --model_name "graph_sentence_extractor" --data_set "ratebeer/medium_500_positive" --data_name "ratebeer_500_positive" --epoch_num 1 --eval --batch_size 1 --learning_rate 0.0001 --optimizer "Adam" --user_embed_size 256 --item_embed_size 256 --feature_embed_size 256 --sent_embed_size 768 --hidden_size 256 --head_num 4 --ffn_inner_hidden_size 256 --ffn_dropout_rate 0.01 --attn_dropout_rate 0.01 --cond_sentence "dcn" --cross_num 2 --cross_type "vector" --dnn_hidden_units 128 128 --model_file "ratebeer_500_positive_graph_sentence_extractor/model_best_12_4_8_46.pt" --eval_output_path "./result/ratebeer_500_pos/top_5/" --select_topk_f 15 --select_topk_s 5