-
Notifications
You must be signed in to change notification settings - Fork 0
/
beijingair.py
63 lines (56 loc) · 1.92 KB
/
beijingair.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
from copent import ci
from knncmi import cmi
from causallearn.utils.KCI.KCI import KCI_CInd
from pycit.estimators import ksg_cmi
from fcit import fcit
from CCIT import CCIT
from pcit.IndependenceTest import PCIT ## the 'mlxtend' package required
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
url = "https://archive.ics.uci.edu/ml/machine-learning-databases/00381/PRSA_data_2010.1.1-2014.12.31.csv"
prsa2010 = pd.read_csv(url)
# index: 5(PM2.5),6(Dew Point),7(Temperature),8(Pressure),10(Cumulative Wind Speed)
varname = ["","","","","","PM2.5","Dew Point","Temperature","Pressure","","CWS"]
eid = 5 # effect
cid = 8 # cause
data = prsa2010.iloc[2200:2700,[eid,cid]].values
te1 = np.zeros(24)
cmi1 = np.zeros(24)
kci1 = np.zeros(24)
ci1 = np.zeros(24)
fcit1 = np.zeros(24)
ccit1 = np.zeros(24)
pcit1 = np.zeros(24)
#pyc1 = np.zeros(24)
for lag in range(1,25):
x1 = data[0:(500-lag),0]
x2 = data[lag:500,0]
y = data[0:(500-lag),1]
## copent
te1[lag-1] = ci(x2,y,x1)
## knncmi
df = pd.DataFrame(np.vstack((x2,y,x1)).T, columns = ['x2','y','x1'])
cmi1[lag-1] = cmi(['x2'], ['y'], ['x1'], 3, df)
## pycit
ci1[lag-1] = ksg_cmi(x2, y, x1)
## fcit
len1 = len(x1)
x2a = np.reshape(x2,[len1,1])
ya = np.reshape(y,[len1,1])
x1a = np.reshape(x1,[len1,1])
fcit1[lag-1] = fcit.test(x2a,ya,x1a)
## causal-learn
kci_cind = KCI_CInd()
_,kci1[lag-1] = kci_cind.compute_pvalue(x2a,ya,x1a)
## CCIT
ccit1[lag-1] = CCIT.CCIT(x2a,ya,x1a)
## pcit
p,tmp,tmp = PCIT(x2a,ya,z = x1a)
pcit1[lag-1] = p[0]
str = "lag: %d; TE:%.3f; CMI:%.3f; KCI:%.3f; CI:%.3f; fcit:%.3f; ccit:%.3f; pcit:%.3f" %(lag,te1[lag-1],cmi1[lag-1],kci1[lag-1],ci1[lag-1],fcit1[lag-1],ccit1[lag-1],pcit1[lag-1])
print(str)
data1 = np.vstack((cmi1,kci1,ci1,fcit1,ccit1,pcit1))
df1 = pd.DataFrame(data1.T)
df1.columns = ["cmi","kci","knn","fcit","ccit","pcit"]
df1.to_csv("~/Rworks/bench/py1.csv")