как построить матрицу корреляции 8x8 - PullRequest
0 голосов
/ 07 октября 2019

Я пытаюсь построить 8x8 correlation matrix между различными характеристиками и соответствующими шансами на признание. Могу ли я узнать, как я должен это сделать?

import tensorflow as tf
import numpy as np
import pylab as plt
from sklearn.model_selection import train_test_split
from sklearn import preprocessing
import pandas as pd

admit_data = np.genfromtxt('admission_predict.csv', delimiter= ',')
X_data, Y_data = admit_data[1:,1:8], admit_data[1:,-1]
x_train, x_test, y_train_, y_test_ = train_test_split(
                                            X_data, 
                                            Y_data, 
                                            test_size=0.3, 
                                            random_state=42
                                      )
scaler = preprocessing.StandardScaler()
x_train = scaler.fit_transform(x_train)
x_test = scaler.fit_transform(x_test)
y_train = y_train_.reshape(len(y_train_), no_labels)
y_test = y_test_.reshape(len(y_test_), no_labels)

data = admit_data
df = pd.DataFrame(data, columns = ['Serial No.','GRE Score','TOEFL Score','University Rating','SOP','LOR','CGPA','Research','Chance of Admit'])
df.corr()

Это код, который я читаю сейчас, и мой файл выглядит вот так

Пожалуйста, помогите мнезаговор 8x8 correlation matrix на данный момент мой код не возвращает 8x8 correlation matrix

Ответы [ 2 ]

1 голос
/ 07 октября 2019

Как насчет

import matplotlib.pyplot as plt
cors = df.corr()
plt.matshow(cors)
plt.yticks(range(cors.shape[1]), cors.columns, fontsize=7)
plt.xticks(range(cors.shape[1]), cors.columns, fontsize=7, rotation=90)
plt.colorbar()

, чтобы использовать все, кроме столбца "Serial No", используйте вместо этого следующие cors:

cors = df.drop("Serial No.", axis=1).corr()
0 голосов
/ 07 октября 2019

Как пример:

import numpy as np
import matplotlib.pyplot as plt

one = np.random.rand(80).reshape(8, 10)
corr = np.corrcoef(one)

fig, ax = plt.subplots(nrows=1, ncols=1, figsize=(6, 6))
mat = ax.imshow(corr)
fig.colorbar(mat, ax=ax, shrink=0.8)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...