cartodb4/crankshaft
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add types to arrays

Browse Source
This commit is contained in:
Andy Eschbacher 2016-03-23 15:33:54 -04:00
parent 42e760b5d1
commit c5a58f97ec
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/py/crankshaft/crankshaft/space_time_dynamics/markov.py
View File

@ -140,9 +140,9 @@ def get_prob_stats(prob_dist, unit_indices):
"""
num_elements = len(prob_dist)
trend_up = np.empty(num_elements)
trend_down = np.empty(num_elements)
trend = np.empty(num_elements)
trend_up = np.empty(num_elements, dtype=float)
trend_down = np.empty(num_elements, dtype=float)
trend = np.empty(num_elements, dtype=float)
for i in range(num_elements):
trend_up[i] = prob_dist[i, (unit_indices[i]+1):].sum()