Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

env.yaml 2.2 KB

You have to be logged in to leave a comment. Sign In
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
  1. paperswithcode_path: "data/paperswithcode_with_tasks.csv"
  2. metrics:
  3. information_retrieval_metric: "accuracy@10"
  4. fasttext:
  5. dimension: 200
  6. epochs: 5
  7. word2vec:
  8. epochs: 1
  9. dimension: 200
  10. gnn:
  11. unsupervised_params:
  12. epochs: 1
  13. num_neighbors: [25, 5]
  14. model_name: "graphsage"
  15. layers: 2
  16. batch_size: 4
  17. hidden_channels: 512
  18. model_name: "graphsage"
  19. lr: 0.001
  20. weight_decay: 0.0001
  21. sentence_transformer_model_name: "flax-sentence-embeddings/st-codesearch-distilroberta-base"
  22. similarity_model_params:
  23. epochs: 5
  24. hidden_channels: 128
  25. batch_size: 16
  26. test_run: False
  27. graph_records_path: "data/dependency_records.csv"
  28. python_files_path: "data/all_crawled_python_files.parquet"
  29. rnn_config:
  30. model_type: "sru"
  31. n_hidden: 128
  32. pooling_mode_max_tokens: True
  33. pooling_mode_mean_tokens: True
  34. tokenizer: "PythonCodeTokenizer"
  35. max_seq_length: 128
  36. transformer:
  37. T5_model: "Salesforce/codet5-base-multi-sum"
  38. # "SEBIS/code_trans_t5_base_source_code_summarization_python_multitask_finetune"
  39. max_source_length: 256
  40. max_target_length: 64
  41. summarization:
  42. transformer_model_name: "doc2query/stackexchange-title-body-t5-base-v1"
  43. ulmfit_path: "output/models/learn_bigger_data_longer.pkl"
  44. nbow:
  45. additional_columns: ["titles", "readme"]
  46. n_readme_lines: 10
  47. config_names: [
  48. 'readme#dependencies-MiniLM-nbow-mnrl',
  49. 'readme-MiniLM-nbow-mnrl',
  50. 'titles-nbow-nbow-mnrl',
  51. 'readme-nbow-nbow-mnrl',
  52. 'titles#dependencies-MiniLM-nbow-mnrl',
  53. 'dependencies-MiniLM-nbow-mse',
  54. 'titles-MiniLM-nbow-mse',
  55. 'titles#dependencies-MiniLM-nbow-mse',
  56. 'titles-MiniLM-nbow-mnrl',
  57. 'dependencies-MiniLM-nbow-mnrl',
  58. 'dependencies-nbow-nbow-mnrl',
  59. 'readme#dependencies-nbow-nbow-mnrl',
  60. 'titles-MiniLM-MiniLM-mnrl',
  61. 'titles#dependencies-nbow-nbow-mnrl',
  62. 'titles-MiniLM-MiniLM-mse',
  63. 'readme#signatures-MiniLM-nbow-mnrl',
  64. 'titles#signatures-MiniLM-nbow-mnrl',
  65. 'signatures-MiniLM-nbow-mse',
  66. 'titles#signatures-MiniLM-nbow-mse',
  67. 'signatures-MiniLM-nbow-mnrl',
  68. 'signatures-nbow-nbow-mnrl',
  69. 'readme#signatures-nbow-nbow-mnrl',
  70. 'titles#signatures-nbow-nbow-mnrl'
  71. ]
  72. neptune_config_path: "neptune_stuff.json"
Tip!

Press p or to see the previous file or, n or to see the next file

Comments

Loading...