|
import pandas as pd |
|
import seaborn as sns |
|
import streamlit as st |
|
import matplotlib.pyplot as plt |
|
import plotly.graph_objects as go |
|
from sklearn.cluster import KMeans |
|
df = pd.read_csv("Spotify-2000.csv") |
|
|
|
|
|
|
|
st.title('Music Genres Clustering :saxophone:') |
|
st.image('https://www.cnet.com/a/img/resize/034ecacea4dbf0ae76529f37c2d0309b17b557ec/hub/2021/11/10/ab5e2d3b-9a4a-41f0-b2cd-6cee804ce823/genre-charts-covers.png?auto=webp&fit=crop&height=675&width=1200') |
|
st.write('**Data Preview:**') |
|
df = df.drop("Index", axis=1) |
|
st.dataframe(df) |
|
|
|
|
|
st.write('**Visualizing Top 5 Genres:**') |
|
top_genres = df["Top Genre"].value_counts().nlargest(5).index |
|
def plot_3d_scatter(df): |
|
fig = go.Figure() |
|
for i in top_genres: |
|
fig.add_trace(go.Scatter3d(x=df[df["Top Genre"]==i]['Beats Per Minute (BPM)'], |
|
y=df[df["Top Genre"]==i]['Energy'], |
|
z=df[df["Top Genre"]==i]['Danceability'], |
|
mode='markers', marker=dict(size=6, line=dict(width=1)), name=str(i))) |
|
fig.update_traces(hovertemplate='BPM: %{x} <br>Energy: %{y} <br>Danceability: %{z}') |
|
fig.update_layout(autosize=True, scene=dict(xaxis_title='BPM', yaxis_title='Energy', zaxis_title='Danceability')) |
|
return fig |
|
st.plotly_chart(plot_3d_scatter(df)) |
|
|
|
|
|
|
|
st.write('**Clustering by "Beats Per Minute (BPM)", "Loudness (dB)", "Liveness", "Valence", "Acousticness","Speechiness" using Kmeans.:**') |
|
df2 = df[["Beats Per Minute (BPM)", "Loudness (dB)", "Liveness", "Valence", "Acousticness", "Speechiness"]] |
|
kmeans = KMeans(n_clusters=10) |
|
clusters = kmeans.fit_predict(df2) |
|
df["Music Segments"] = clusters |
|
df["Music Segments"] = df["Music Segments"].map({i: f"Cluster {i+1}" for i in range(10)}) |
|
df["Music Segments"].fillna("Cluster 10", inplace=True) |
|
st.dataframe(df) |
|
|
|
st.write('**Visualizing:**') |
|
def plot_3d_scatter(df): |
|
fig = go.Figure() |
|
for i in df["Music Segments"].unique(): |
|
fig.add_trace(go.Scatter3d(x=df[df["Music Segments"]==i]['Beats Per Minute (BPM)'], |
|
y=df[df["Music Segments"]==i]['Energy'], |
|
z=df[df["Music Segments"]==i]['Danceability'], |
|
mode='markers', marker=dict(size=6, line=dict(width=1)), name=str(i))) |
|
fig.update_traces(hovertemplate='BPM: %{x} <br>Energy: %{y} <br>Danceability: %{z}') |
|
fig.update_layout(autosize=True, scene=dict(xaxis_title='BPM', yaxis_title='Energy', zaxis_title='Danceability')) |
|
return fig |
|
st.plotly_chart(plot_3d_scatter(df)) |
|
|
|
|
|
|
|
st.write('**Similarity between Top 5 Genres and Clusters:**') |
|
genre_cluster_df = pd.crosstab(df["Top Genre"], df["Music Segments"]) |
|
|
|
|
|
genre_cluster_df = genre_cluster_df.loc[top_genres] |
|
|
|
|
|
scatter_data = genre_cluster_df.reset_index().melt(id_vars="Top Genre", var_name="Cluster", value_name="Frequency") |
|
scatter_data["Cluster"] = scatter_data["Cluster"].apply(lambda x: int(x.split()[1]) if x != "Cluster 10" else 10) |
|
|
|
|
|
plt.figure(figsize=(12, 8)) |
|
sns.scatterplot(data=scatter_data, x="Cluster", y="Frequency", hue="Top Genre", palette="Set1", s=100) |
|
plt.xlabel("Cluster") |
|
plt.ylabel("Frequency") |
|
plt.legend(title="Top Genre", bbox_to_anchor=(1.05, 1), loc='upper left') |
|
plt.savefig("scatter_plot.png", bbox_inches='tight') |
|
st.pyplot(plt) |
|
|