Spaces:

Circhastic
/

Code-Readability-Classifier

Sleeping

Circhastic commited on Mar 10

Commit

4c6b685

1 Parent(s): bf17004

updated model threshold

Files changed (1) hide show

app.py CHANGED Viewed

@@ -39,7 +39,7 @@ def classify_code(input_text, input_file):
     tokenized_code = tokenize_java_code(cleaned_code)
     prediction = model.predict(tokenized_code)[0][0]
-    threshold = 0.51 # Increase the threshold for "Readable"
     prediction = (prediction > threshold).astype(int)  # Convert probabilities to binary
     # Convert to readable/unreadable

     tokenized_code = tokenize_java_code(cleaned_code)
     prediction = model.predict(tokenized_code)[0][0]
+    threshold = 0.52 # Increase the threshold for "Readable"
     prediction = (prediction > threshold).astype(int)  # Convert probabilities to binary
     # Convert to readable/unreadable