2350e18231a67fa9d795fe7a16919b344a4b4ce5,examples/uv/UV_correlations.py,,,#,17

Before Change

y_train = train_dataset.y
n_tasks = y_train.shape[1]

all_results = []
for task in range(n_tasks):
  y_task = y_train[:, task]
  task_results = []
  for other_task in range(n_tasks):
    if task == other_task:
      task_results.append(1.)
      continue
    y_other = y_train[:, other_task]
    r2 = dc.metrics.pearson_r2_score(y_task, y_other)
    print("r2 for %s-%s is %f" % (task, other_task, r2))
    task_results.append(r2)
  print("Task %d" % task)
  print(task_results)
  all_results.append(task_results)
print("Writing results to uv_corr.csv")
df = pd.DataFrame(all_results)
df.to_csv("uv_corr.csv")

After Change

import pandas as pd
import matplotlib
// Force matplotlib to not use any Xwindows backend.
matplotlib.use("Agg")
import matplotlib.mlab as mlab
import matplotlib.pyplot as plt
from UV_datasets import load_uv

Instances