ItemStudy/main.py
2024-05-15 22:10:34 +08:00

33 lines
586 B
Python

import file_load
import json
import os
import numpy
# file_load.make_data()
similarity = file_load.similarity()
correlation = file_load.correlation()
x=[]
y=[]
s="similarity, correlation\n"
for i in similarity:
try:
s+=str(i["similarity"])+','+str(correlation[i["from"],i["to"]])+'\n'
x.append(i["similarity"])
y.append(correlation[i["from"],i["to"]])
if(i['similarity']>0.5 and correlation[i['from'],i['to']]<-0.5):
print(i,correlation[i['from'],i['to']])
except:
pass
print(numpy.corrcoef(x,y)[0,1])
with open("Work/point.csv","w") as point:
point.write(s)