imamnurby commited on
Commit
e34bb70
1 Parent(s): a0f8976

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +6 -6
config.py CHANGED
@@ -3,14 +3,14 @@ config = {
3
  'tokenizer_path_retrieval': 'imamnurby/bow-tokenizer-uncased',
4
 
5
  # model_path
6
- 'model_path_retrieval': 'library_retriever/retrieval_distillbert',
7
- 'model_path_generative': 'pattern_generator/generative_codebert2codebert',
8
- 'model_path_classifier': 'config_classifier/classifier_codebert_11nov_latest',
9
- 'classifier_head_path': 'config_classifier/classifier_head_rf_11nov.pkl',
10
 
11
  # db path
12
- 'db_metadata_path': '.assets/lib_to_features.csv',
13
- 'db_constructor_path': 'assets/lib_to_constructor.csv',
14
 
15
  # retrieval_model_setting
16
  'max_k': 10,
 
3
  'tokenizer_path_retrieval': 'imamnurby/bow-tokenizer-uncased',
4
 
5
  # model_path
6
+ 'model_path_retrieval': './library_retriever/retrieval_distillbert',
7
+ 'model_path_generative': './pattern_generator',
8
+ 'model_path_classifier': './configuration_classifier',
9
+ 'classifier_head_path': './configuration_classifier/classifier_head_rf_11nov.pkl',
10
 
11
  # db path
12
+ 'db_metadata_path': '.lib_to_features.csv',
13
+ 'db_constructor_path': '.lib_to_constructor.csv',
14
 
15
  # retrieval_model_setting
16
  'max_k': 10,