age_prediction_graph_validation_set.py 571 Bytes
# -*- coding: utf-8 -*-
"""age-prediction_graph_validation_set.ipynb

Automatically generated by Colaboratory.

Original file is located at
    https://colab.research.google.com/drive/16Rn0IDJTE6vJMOZtafDAI19AvGD9eK4C
"""

import pandas as pd
import matplotlib.pyplot as plt

plt.xlabel("Age")
plt.ylabel("Prediction")

df = pd.read_csv('/content/drive/MyDrive/2d-slice-set-networks-for-brain-age-master/data/valid_pred.csv')
df1 = pd.DataFrame(df).T
corr = df.corr(method='pearson')
print(corr)

X=df["Age"]
y=df["Prediction"]

plt.plot(X, y, 'o')
plt.plot(y,y)
plt.show