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:34:58 -03:00 committed by GitHub
parent c4f264a1f2
commit 6cc88f6292
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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