wip
This commit is contained in:
13
votesits.py
13
votesits.py
@@ -8,13 +8,12 @@ from datetime import datetime
|
||||
from datetime import timedelta
|
||||
from dateutil.relativedelta import relativedelta
|
||||
|
||||
from common import calc_intervals, printnoln, rprint, DAYS_NEW_USER, FIG_SIZE, difftime
|
||||
from common import calc_intervals, printnoln, rprint, DAYS_NEW_USER, FIG_SIZE, CHANGE_DATE, difftime
|
||||
from loader import load, dmt, cms, readVotes
|
||||
from sentiments import readtoxleveltxt
|
||||
|
||||
colors = ['red', 'green', 'blue', 'orange', 'deeppink']
|
||||
thresholds = [6, 9, 12, 15]
|
||||
changedate = datetime.fromisoformat("2018-09-01T00:00:00")
|
||||
|
||||
|
||||
def main(folder, intervl):
|
||||
@@ -76,7 +75,7 @@ def main(folder, intervl):
|
||||
|
||||
print("Computing full ITS")
|
||||
t = np.reshape(np.array([i for i in range(len(datasingle)) for j in datasingle[i]]), (-1, 1))
|
||||
x = np.reshape(np.array([(0 if intervals[i][0] <= changedate else 1) for i in range(len(datasingle)) for j in datasingle[i]]), (-1, 1))
|
||||
x = np.reshape(np.array([(0 if intervals[i][0] <= CHANGE_DATE else 1) for i in range(len(datasingle)) for j in datasingle[i]]), (-1, 1))
|
||||
X = np.array(t)
|
||||
X = np.concatenate((X, x), 1)
|
||||
X = np.concatenate((X, np.multiply(t, x)), 1)
|
||||
@@ -97,14 +96,14 @@ def main(folder, intervl):
|
||||
thresp = []
|
||||
print("Computing threshold ITS")
|
||||
for ti in thresholds:
|
||||
# print(1, changedate - relativedelta(months=ti))
|
||||
# print(2, changedate + relativedelta(months=ti))
|
||||
z = [(i, x) for (i, x) in zip(intervals, datasingle) if i[0] >= changedate - relativedelta(months=ti) and i[1] <= changedate + relativedelta(months=ti)]
|
||||
# print(1, CHANGE_DATE - relativedelta(months=ti))
|
||||
# print(2, CHANGE_DATE + relativedelta(months=ti))
|
||||
z = [(i, x) for (i, x) in zip(intervals, datasingle) if i[0] >= CHANGE_DATE - relativedelta(months=ti) and i[1] <= CHANGE_DATE + relativedelta(months=ti)]
|
||||
iv = [i for (i, x) in z]
|
||||
# print("iv " + str(iv))
|
||||
d = [x for (i, x) in z]
|
||||
t = np.reshape(np.array([i for i in range(len(d)) for j in d[i]]), (-1, 1))
|
||||
x = np.reshape(np.array([(0 if iv[i][0] <= changedate else 1) for i in range(len(d)) for j in d[i]]), (-1, 1))
|
||||
x = np.reshape(np.array([(0 if iv[i][0] <= CHANGE_DATE else 1) for i in range(len(d)) for j in d[i]]), (-1, 1))
|
||||
X = np.array(t)
|
||||
X = np.concatenate((X, x), 1)
|
||||
X = np.concatenate((X, np.multiply(t, x)), 1)
|
||||
|
||||
Reference in New Issue
Block a user