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

n, bins, patches = plt.hist(np.array(all_results), 50, normed=True, stacked=True,
                            facecolor="green", alpha=0.75)
plt.xlabel("Cross-task Correlations")
plt.ylabel("Probability Density")
plt.title("Histogram of UV Intertask Correlations")
plt.grid(True)
plt.savefig("UV_correlations.png")

Instances