python中visuals模块_visuals.py · 肥宅宇/Machine-Learning-with-Python - Gitee.com

###########################################

# Suppress matplotlib user warnings

# Necessary for newer version of matplotlib

import warnings

warnings.filterwarnings("ignore", category = UserWarning, module = "matplotlib")

#

# Display inline matplotlib plots with IPython

from IPython import get_ipython

get_ipython().run_line_magic('matplotlib', 'inline')

###########################################

import matplotlib.pyplot as plt

import matplotlib.cm as cm

import pandas as pd

import numpy as np

def pca_results(good_data, pca):

'''

Create a DataFrame of the PCA results

Includes dimension feature weights and explained variance

Visualizes the PCA results

'''

# Dimension indexing

dimensions = dimensions = ['Dimension {}'.format(i) for i in range(1,len(pca.components_)+1)]

# PCA components

components = pd.DataFrame(np.round(pca.components_, 4), columns = good_data.keys())

components.index = dimensions

# PCA explained variance

ratios = pca.explained_variance_ratio_.reshape(len(pca.components_), 1)

variance_ratios = pd.DataFrame(np.round(ratios, 4), columns = ['Explained Variance'])

variance_ratios.index = dimensions

# Create a bar plot v