mirror of
https://github.com/msoedov/agentic_security.git
synced 2026-07-07 12:17:48 +02:00
fix(warnings):
This commit is contained in:
@@ -1,9 +1,11 @@
|
||||
import importlib.resources as pkg_resources
|
||||
import os
|
||||
import warnings
|
||||
|
||||
import joblib
|
||||
import pandas as pd
|
||||
from sklearn.feature_extraction.text import TfidfVectorizer
|
||||
from sklearn.exceptions import InconsistentVersionWarning
|
||||
from sklearn.preprocessing import StandardScaler
|
||||
from sklearn.svm import OneClassSVM
|
||||
|
||||
@@ -70,27 +72,33 @@ class RefusalClassifier:
|
||||
"""
|
||||
Load the trained model, vectorizer, and scaler from disk.
|
||||
"""
|
||||
try:
|
||||
self.model = joblib.load(self.model_path)
|
||||
self.vectorizer = joblib.load(self.vectorizer_path)
|
||||
self.scaler = joblib.load(self.scaler_path)
|
||||
except FileNotFoundError:
|
||||
# Load from package resources
|
||||
package = (
|
||||
__package__ # This should be 'agentic_security.refusal_classifier'
|
||||
)
|
||||
with warnings.catch_warnings():
|
||||
warnings.filterwarnings("ignore", category=InconsistentVersionWarning)
|
||||
try:
|
||||
self.model = joblib.load(self.model_path)
|
||||
self.vectorizer = joblib.load(self.vectorizer_path)
|
||||
self.scaler = joblib.load(self.scaler_path)
|
||||
except FileNotFoundError:
|
||||
# Load from package resources
|
||||
package = (
|
||||
__package__ # This should be 'agentic_security.refusal_classifier'
|
||||
)
|
||||
|
||||
# Load model
|
||||
with pkg_resources.open_binary(package, "oneclass_svm_model.joblib") as f:
|
||||
self.model = joblib.load(f)
|
||||
# Load model
|
||||
with pkg_resources.open_binary(
|
||||
package, "oneclass_svm_model.joblib"
|
||||
) as f:
|
||||
self.model = joblib.load(f)
|
||||
|
||||
# Load vectorizer
|
||||
with pkg_resources.open_binary(package, "tfidf_vectorizer.joblib") as f:
|
||||
self.vectorizer = joblib.load(f)
|
||||
# Load vectorizer
|
||||
with pkg_resources.open_binary(
|
||||
package, "tfidf_vectorizer.joblib"
|
||||
) as f:
|
||||
self.vectorizer = joblib.load(f)
|
||||
|
||||
# Load scaler
|
||||
with pkg_resources.open_binary(package, "scaler.joblib") as f:
|
||||
self.scaler = joblib.load(f)
|
||||
# Load scaler
|
||||
with pkg_resources.open_binary(package, "scaler.joblib") as f:
|
||||
self.scaler = joblib.load(f)
|
||||
|
||||
def is_refusal(self, text):
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user