Voice-Emotion / backend /evaluate_model.py
aniruddhakumarpaul's picture
Upload folder using huggingface_hub
f74654d verified
import os
import sys
import numpy as np
import pickle
from sklearn.model_selection import train_test_split
from sklearn.metrics import classification_report, accuracy_score, f1_score
from tensorflow.keras.models import load_model
from tensorflow.keras.utils import to_categorical
# Data Paths
BASE_DIR = os.path.dirname(os.path.abspath(__file__)) # This is backend/
ROOT_DIR = os.path.dirname(BASE_DIR) # This is project root
DATA_DIR = os.path.join(ROOT_DIR, "data sets")
MODEL_PATH = os.path.join(ROOT_DIR, "model.h5")
ENCODER_PATH = os.path.join(ROOT_DIR, "encoder.pkl")
FEATURES_PATH = os.path.join(DATA_DIR, "features_cache.npy")
LABELS_PATH = os.path.join(DATA_DIR, "labels_cache.npy")
def evaluate():
print("Loading data from cache...")
if not os.path.exists(FEATURES_PATH) or not os.path.exists(LABELS_PATH):
print("Error: Cached features not found. Please train the model first.")
return
X = np.load(FEATURES_PATH)
y = np.load(LABELS_PATH)
print(f"Loaded {len(X)} samples.")
print("Loading Label Encoder...")
with open(ENCODER_PATH, 'rb') as f:
le = pickle.load(f)
# Encode labels (Same logic as training)
y_encoded = to_categorical(le.fit_transform(y))
# Split (Same random_state as training to ensure same test set)
print("Splitting data (random_state=42)...")
X_train, X_test, y_train, y_test = train_test_split(X, y_encoded, test_size=0.2, random_state=42, stratify=y)
print(f"Test Set Size: {len(X_test)}")
print("Loading Model...")
model = load_model(MODEL_PATH)
print("Evaluating...")
# Predict
y_pred_prob = model.predict(X_test, verbose=0)
y_pred = np.argmax(y_pred_prob, axis=1)
y_true = np.argmax(y_test, axis=1)
# Calculate Metrics
accuracy = accuracy_score(y_true, y_pred)
f1 = f1_score(y_true, y_pred, average='weighted')
print("\n" + "="*30)
print(f"Accuracy: {accuracy:.4f}")
print(f"F1 Score (Weighted): {f1:.4f}")
print("="*30 + "\n")
# Detailed Report
target_names = le.classes_
print("Classification Report:")
print(classification_report(y_true, y_pred, target_names=target_names))
if __name__ == "__main__":
evaluate()