result_matrix[first_header][second_header])
print("Pearson correlation coefficients have calculated!")
return result_matrix
def write_to_csv(file_name, headers, data):
with open(file_name, 'w') as outfile:
# the first cell is empty
outfile.write(";" + ";".join(headers) + "\n")
for first_header in headers:
outfile.write(first_header)
for second_header in headers:
outfile.write(";" + str(data[first_header][second_header][0]) +
" " + str(data[first_header][second_header][1]))
outfile.write("\n")
data_handler = DataHandler()
signals, backgrounds = data_handler.get_separate_training_data()
headers = data_handler.get_headers()
# signals_backgrounds_correlation(headers, signals, backgrounds)
correlation_matrix_signals = pearson_correlation_matrix(headers, signals)
# file_name = "./reports/correlation/Pearson_correlation_signals.csv"
# write_to_csv(file_name, headers, correlation_matrix_signals)
# correlation_matrix_backgrounds = pearson_correlation_matrix(headers, backgrounds)
# file_name = "./reports/correlation/Pearson_correlation_backgrounds.csv"
# write_to_csv(file_name, headers, correlation_matrix_backgrounds)
