batik-classifier / convert_models.py
maftuh-main's picture
Remove old KNN model files, keep only Keras model
f5ede97
import pickle
import joblib
import sys
print("Loading models with current environment...")
print("numpy version:", sys.version)
# Load models
knn_model = joblib.load("models/batik_knn_model_95acc.pkl")
label_encoder = joblib.load("models/batik_classes.pkl")
scaler = joblib.load("models/scaler.joblib")
metadata = joblib.load("models/batik_model_metadata.pkl")
print(" Models loaded successfully")
print("Re-saving with protocol=4 for compatibility...")
# Re-save with older pickle protocol
joblib.dump(knn_model, "models/batik_knn_model_95acc_compat.pkl", protocol=4)
joblib.dump(label_encoder, "models/batik_classes_compat.pkl", protocol=4)
joblib.dump(scaler, "models/scaler_compat.joblib", protocol=4)
joblib.dump(metadata, "models/batik_model_metadata_compat.pkl", protocol=4)
print(" Models re-saved with protocol=4")
print("Testing load...")
# Test load
test_knn = joblib.load("models/batik_knn_model_95acc_compat.pkl")
print(" Compatible models created successfully!")