Upload prediction.py

prediction.py

@@ -74,6 +74,14 @@ def run():
                 # y_cluster = []
                 #for rd in range(0, len(df_churn)): y_cluster.append(random.randint(0, 2)) # Random Generator for testing
                 df_churn['cluster'] = y_cluster
+
+                temp_cols = df_churn.columns.tolist()
+                new_cols = temp_cols[0:1] + temp_cols[-2:] + temp_cols[1:-2]
+                df_churn = df_churn[new_cols]
+                df_churn = df_churn.sort_values(by=['cluster'], ascending=True)
+
+                # Saving Result to Excel
+                df_churn.to_excel('model_result.xlsx', index=False)
                 
                 # Split Data into 3 Cluster DataFrames
                 df_cluster_0 = df_churn[df_churn['cluster'] == 0]
@@ -156,8 +164,18 @@ def run():
                     ax.set_ylabel(f'Average {column}')
                     ax.bar_label(ax.containers[0])
                 
+                # plt.style.use('dark_background')
                 plt.tight_layout()
                 st.pyplot(fig)
+                # plt.style.use('default')
+
+                with open('model_result.xlsx', 'rb') as file:
+                    st.download_button(
+                        label='💾 Download Prediction Result',
+                        data=file,
+                        file_name='model_result.xlsx',
+                        mime='application/vnd.ms-excel'
+                )
     
     def tenureMonthToYear():
         year = st.session_state.tenurem % 12