Я могу запустить простой пример фильтра Калмана pykalman, приведенный в документации pykalman:
import pykalman
import numpy as np
kf = pykalman.KalmanFilter(transition_matrices = [[1, 1], [0, 1]], observation_matrices = [[0.1, 0.5], [-0.3, 0.0]])
measurements = np.asarray([[1,0], [0,0], [0,1]]) # 3 observations
(filtered_state_means, filtered_state_covariances) = kf.filter(measurements)
print filtered_state_means
Это правильно возвращает оценки состояния (по одной для каждого наблюдения):
[[ 0.07285974 0.39708561]
[ 0.30309693 0.2328318 ]
[-0.5533711 -0.0415223 ]]
Однако, если я предоставлю только одно наблюдение, код завершится ошибкой:
import pykalman
import numpy as np
kf = pykalman.KalmanFilter(transition_matrices = [[1, 1], [0, 1]], observation_matrices = [[0.1, 0.5], [-0.3, 0.0]])
measurements = np.asarray([[1,0]]) # 1 observation
(filtered_state_means, filtered_state_covariances) = kf.filter(measurements)
print filtered_state_means
со следующей ошибкой:
ValueError: could not broadcast input array from shape (2,2) into shape (2,1)
Как я могу использовать pykalman для обновления начального состояния и начальной ковариации, используя всего одно наблюдение?