Bugfix in Measures.get_point_statistics for multivariate models

This commit is contained in:
Petrônio Cândido de Lima e Silva 2018-06-28 14:31:30 -03:00 committed by GitHub
parent 1a00a4aa4e
commit c4f264a1f2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -324,6 +324,9 @@ def get_point_statistics(data, model, **kwargs):
if steps_ahead == 1:
forecasts = model.predict(ndata, **kwargs)
if model.is_multivariate:
ndata = ndata[model1.target_variable.data_label].values
if not isinstance(forecasts, (list, np.ndarray)):
forecasts = [forecasts]