def train_transform_func(vector):
# Remap the value to elimiate NAN result for ease of calculation
new_vec = DenseVector.toArray().map(lambda x: 0 if math.isnan(x) else x)
arimaModel = ARIMA.fit_model(1, 0, 0, new_vec)
forecasted = arimaModel.forecast(new_vec, 5) # 5 days for predict
print(type(forecasted))
exit(0)
