Coverage for hopwise/utils/argument_list.py: 100%

1# @Time : 2020/10/19 

2# @Author : Shanlei Mu 

3# @Email : slmu@ruc.edu.cn 

4 

5# yapf: disable 

6 

7general_arguments = [ 

8 'gpu_id', 'use_gpu', 

9 'seed', 

10 'reproducibility', 

11 'state', 

12 'data_path', 

13 'checkpoint_dir', 

14 'show_progress', 

15 'config_file', 

16 'save_dataset', 

17 'dataset_save_path', 

18 'save_dataloaders', 

19 'dataloaders_save_path', 

20 'log_wandb', 

21] 

22 

23training_arguments = [ 

24 'epochs', 'train_batch_size', 

25 'learner', 'learning_rate', 

26 'train_neg_sample_args', 

27 'eval_step', 'stopping_step', 

28 'clip_grad_norm', 

29 'weight_decay', 

30 'loss_decimal_place', 

31] 

32 

33evaluation_arguments = [ 

34 'eval_args', 'repeatable', 

35 'metrics', 'topk', 'valid_metric', 'valid_metric_bigger', 

36 'eval_batch_size', 

37 'metric_decimal_place', 

38] 

39 

40dataset_arguments = [ 

41 'field_separator', 'seq_separator', 

42 'USER_ID_FIELD', 'ITEM_ID_FIELD', 'RATING_FIELD', 'TIME_FIELD', 

43 'seq_len', 

44 'LABEL_FIELD', 'threshold', 

45 'NEG_PREFIX', 

46 'ITEM_LIST_LENGTH_FIELD', 'LIST_SUFFIX', 'MAX_ITEM_LIST_LENGTH', 'POSITION_FIELD', 

47 'HEAD_ENTITY_ID_FIELD', 'TAIL_ENTITY_ID_FIELD', 'RELATION_ID_FIELD', 'ENTITY_ID_FIELD', 

48 'load_col', 'unload_col', 'unused_col', 'additional_feat_suffix', 

49 'rm_dup_inter', 'val_interval', 'filter_inter_by_user_or_item', 

50 'user_inter_num_interval', 'item_inter_num_interval', 

51 'alias_of_user_id', 'alias_of_item_id', 'alias_of_entity_id', 'alias_of_relation_id', 

52 'preload_weight', 'normalize_field', 'normalize_all', 

53 'benchmark_filename', 

54]