mirror of
https://github.com/SkalaraAI/management-llm.git
synced 2025-04-09 15:00:19 -04:00
29 lines
864 B
Python
29 lines
864 B
Python
import joblib
|
|
|
|
# To be translated
|
|
sample_text = "Implement a login page with form validation using React.js."
|
|
|
|
# Loads the model, vectorizer, and encoder
|
|
model_filename = "svm_model.sav"
|
|
svm_model = joblib.load(model_filename)
|
|
|
|
vectorizer_filename = 'fitted_vectorizer.joblib'
|
|
word_vectorizer = joblib.load(vectorizer_filename)
|
|
|
|
encoder_filename = 'fitted_encoder.joblib'
|
|
encoder = joblib.load(encoder_filename)
|
|
|
|
# Vectorize
|
|
vectorized_text = word_vectorizer.transform([sample_text])
|
|
|
|
binary_predictions = svm_model.predict(vectorized_text)
|
|
|
|
decision_function_scores = svm_model.decision_function(vectorized_text)
|
|
|
|
# Get the top 3 predicted labels based on highest decision function scores
|
|
top_4_indices = decision_function_scores.argsort()[0][-3:][::-1]
|
|
|
|
top_4_labels = encoder.inverse_transform(top_4_indices)
|
|
|
|
print("Top 3 predicted labels:", top_4_labels)
|