Open xiliu-m opened 5 years ago
mean_test_acc: 0.6740, mean_test_f1: 0.6105
parser.add_argument('--model_name', default='new6', type=str) parser.add_argument('--dataset', default='laptop', type=str, help='twitter, restaurant, laptop') parser.add_argument('--optimizer', default='adam', type=str) parser.add_argument('--initializer', default='xavieruniform', type=str) parser.add_argument('--learning_rate', default=1e-3, type=float, help='try 5e-5, 2e-5 for BERT, 1e-3 for others') parser.add_argument('--dropout', default=0.1, type=float) parser.add_argument('--l2reg', default=0.00001, type=float) parser.add_argument('--num_epoch', default=30, type=int, help='try larger number for non-BERT models') parser.add_argument('--batch_size', default=64, type=int, help='try 16, 32, 64 for BERT models') parser.add_argument('--log_step', default=10, type=int) parser.add_argument('--embed_dim', default=300, type=int) parser.add_argument('--hidden_dim', default=300, type=int) parser.add_argument('--bert_dim', default=768, type=int) parser.add_argument('--pretrained_bert_name', default='bert-base-uncased', type=str) parser.add_argument('--max_seq_len', default=80, type=int) parser.add_argument('--polarities_dim', default=3, type=int) parser.add_argument('--hops', default=3, type=int) parser.add_argument('--device', default=None, type=str, help='e.g. cuda:0') parser.add_argument('--seed', default=0, type=int, help='set seed for reproducibility') parser.add_argument('--cross_val_fold', default=10, type=int, help='k-fold cross validation')
mean_test_acc: 0.6740, mean_test_f1: 0.6105
parser.add_argument('--model_name', default='new6', type=str) parser.add_argument('--dataset', default='laptop', type=str, help='twitter, restaurant, laptop') parser.add_argument('--optimizer', default='adam', type=str) parser.add_argument('--initializer', default='xavieruniform', type=str) parser.add_argument('--learning_rate', default=1e-3, type=float, help='try 5e-5, 2e-5 for BERT, 1e-3 for others') parser.add_argument('--dropout', default=0.1, type=float) parser.add_argument('--l2reg', default=0.00001, type=float) parser.add_argument('--num_epoch', default=30, type=int, help='try larger number for non-BERT models') parser.add_argument('--batch_size', default=64, type=int, help='try 16, 32, 64 for BERT models') parser.add_argument('--log_step', default=10, type=int) parser.add_argument('--embed_dim', default=300, type=int) parser.add_argument('--hidden_dim', default=300, type=int) parser.add_argument('--bert_dim', default=768, type=int) parser.add_argument('--pretrained_bert_name', default='bert-base-uncased', type=str) parser.add_argument('--max_seq_len', default=80, type=int) parser.add_argument('--polarities_dim', default=3, type=int) parser.add_argument('--hops', default=3, type=int) parser.add_argument('--device', default=None, type=str, help='e.g. cuda:0') parser.add_argument('--seed', default=0, type=int, help='set seed for reproducibility') parser.add_argument('--cross_val_fold', default=10, type=int, help='k-fold cross validation')