Update README.md
Browse files
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/
|
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;
|