Mercurial Hosting > traffic-intelligence
changeset 1124:91faf679e898
minor
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Fri, 29 Nov 2019 00:59:46 -0500 |
parents | 0548a78852b8 |
children | 2682b4696cdf |
files | trafficintelligence/utils.py |
diffstat | 1 files changed, 5 insertions(+), 1 deletions(-) [+] |
line wrap: on
line diff
--- a/trafficintelligence/utils.py Wed Jul 17 17:05:19 2019 -0400 +++ b/trafficintelligence/utils.py Fri Nov 29 00:59:46 2019 -0500 @@ -12,7 +12,7 @@ from scipy.spatial import distance from scipy.sparse import dok_matrix from numpy import zeros, array, exp, sum as npsum, int as npint, arange, cumsum, mean, median, percentile, isnan, ones, convolve, dtype, isnan, NaN, ma, isinf, savez, load as npload, log, polyfit, float as npfloat -from numpy.random import permutation as nppermutation +from numpy.random import random_sample, permutation as nppermutation from pandas import DataFrame, concat import matplotlib.pyplot as plt @@ -305,6 +305,10 @@ # smoothed[i] = 0 # return smoothed +def generateData(nrows, nvariables, scale): + x = random_sample(nrows*nvariables).reshape(nrows,nvariables)*scale + return DataFrame(x, columns=['x{}'.format(i+1) for i in range(nvariables)]) + def kernelSmoothing(x, X, Y, weightFunc, halfwidth): '''Returns the smoothed estimate of (X,Y) at x Sum_x weight(sample_x,x) * y(x)'''