NTUYG commited on
Commit
81cc7fa
1 Parent(s): e799726

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -6,7 +6,7 @@ from simpletransformers.classification import ClassificationModel, Classificatio
6
  name_file = ['bash', 'c', 'c#', 'c++','css', 'haskell', 'java', 'javascript', 'lua', 'objective-c', 'perl', 'php', 'python','r','ruby', 'scala', 'sql', 'swift', 'vb.net']
7
 
8
  deep_scc_model_args = ClassificationArgs(num_train_epochs=10,max_seq_length=300,use_multiprocessing=False)
9
- deep_scc_model = ClassificationModel("roberta", "NTUYG/DeepSCC_RoBERTa", num_labels=19, args=deep_scc_model_args,use_cuda=True)
10
 
11
  code = ''' public static double getSimilarity(String phrase1, String phrase2) {
12
  return (getSC(phrase1, phrase2) + getSC(phrase2, phrase1)) / 2.0;
 
6
  name_file = ['bash', 'c', 'c#', 'c++','css', 'haskell', 'java', 'javascript', 'lua', 'objective-c', 'perl', 'php', 'python','r','ruby', 'scala', 'sql', 'swift', 'vb.net']
7
 
8
  deep_scc_model_args = ClassificationArgs(num_train_epochs=10,max_seq_length=300,use_multiprocessing=False)
9
+ deep_scc_model = ClassificationModel("roberta", "NTUYG/DeepSCC-RoBERTa", num_labels=19, args=deep_scc_model_args,use_cuda=True)
10
 
11
  code = ''' public static double getSimilarity(String phrase1, String phrase2) {
12
  return (getSC(phrase1, phrase2) + getSC(phrase2, phrase1)) / 2.0;