Spaces:
Running
Running
File size: 2,478 Bytes
fc7b4a9 75d43d2 fc7b4a9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
from datetime import datetime
import librosa
import numpy as np
from pathlib import Path
from src.musiclime.explainer import MusicLIMEExplainer
from src.musiclime.wrapper import MusicLIMEPredictor
from src.musiclime.print_utils import green_bold
def explain():
# Start timing and time stamp to record how long the entire explanation thingy is
start_time = datetime.now()
print(
green_bold(
f"[MusicLIME] Started at: {start_time.strftime('%Y-%m-%d %H:%M:%S')}"
)
)
# Create musiclime-related instances
explainer = MusicLIMEExplainer()
predictor = MusicLIMEPredictor()
# Set the path for audio and lyrics [these are samples only - song is Silver Spring]
audio_path = Path("data/external/sample_2.mp3")
lyrics_path = Path("data/external/sample_2.txt")
# Load the audio as an object + load the lyrics as string
y, sr = librosa.load(audio_path)
lyrics_text = lyrics_path.read_text(encoding="utf-8")
# Generate explanations using musiclime
explanation = explainer.explain_instance(
audio=y,
lyrics=lyrics_text,
predict_fn=predictor,
num_samples=5,
labels=(1,),
)
# Get original prediction (first sample is always the orig meaning unperturbed)
original_prediction = explanation.predictions[0]
predicted_class = np.argmax(original_prediction)
# Print explanations
results = explanation.get_explanation(label=1, num_features=10)
print("\n" + "=" * 80)
print(
f"[MusicLIME] Top 10 most important features for {"Human-Composed" if predicted_class == 1 else "AI-Generated"} prediction"
)
print("=" * 80)
for i, item in enumerate(results, 1):
print(
f"#{i:2d} | {item['type']:6s} | {item['feature'][:50]:50s} | weight: {item['weight']:+.6f}"
)
print("=" * 80)
print(f"[MusicLIME] Total features analyzed: {len(results)}")
print("[MusicLIME] Higher absolute weights = more important for the prediction")
# End timing and timestamp
end_time = datetime.now()
total_duration = end_time - start_time
total_minutes = total_duration.total_seconds() / 60
print(f"\n[MusicLIME] Finished at: {end_time.strftime('%Y-%m-%d %H:%M:%S')}")
print(
green_bold(
f"[MusicLIME] Total execution time: {total_minutes:.2f} minutes ({total_duration.total_seconds():.1f} seconds)"
)
)
if __name__ == "__main__":
explain()
|