KNN Wind Power ForecastingΒΆ

K-nearest neighbors for prediction of a time series of wind power for a target turbine in Tehachapi. The prediction model is based on the wind power time series of the target turbine and the time series of its neighbors defined by a radius of 3 kilometers. For the mapping of pattern-label combinations the Power Mapping is used. The power mapping is based on the generaltimeseriesmodel. The feature window and the forecast horizon both consist of 3 elements of every time series. Because of performance issues, in this example only every fifth element is used for training and testing. The plot compares the KNN prediction and the naive model to the actual wind power measurements.

MSE y_hat (KNN-Regressor):  8.20896
MSE naive_hat (Persistence):  9.21491
# Author: Nils A. Treiber <>
# Oliver Kramer <>
# Jendrik Poloczek <>
# Justin P. Heinermann <>
# Stefan Oehmcke <>
# License: BSD 3 clause

import math
import matplotlib.pyplot as plt
from numpy import zeros, float32
from windml.datasets.nrel import NREL
from windml.mapping.power_mapping import PowerMapping
from sklearn.neighbors import KNeighborsRegressor
from sklearn.metrics import mean_squared_error

# get windpark and corresponding target. forecast is for the target turbine
park_id = NREL.park_id['tehachapi']
windpark = NREL().get_windpark(park_id, 3, 2004, 2005)
target = windpark.get_target()

# use power mapping for pattern-label mapping.
feature_window, horizon = 3,3
mapping = PowerMapping()
X = mapping.get_features_park(windpark, feature_window, horizon)
y = mapping.get_labels_turbine(target, feature_window, horizon)

# train roughly for the year 2004, test roughly for the year 2005.
train_to,test_to = int(math.floor(len(X) * 0.5)), len(X)
# train and test only every fifth pattern, for performance.
train_step, test_step = 5, 5

# initialize and fit a KNN regressor from sklearn with k=10 neighbors.
reg = KNeighborsRegressor(10, 'uniform').fit(X_train,y_train)
# run the knn regression
y_hat = reg.predict(X_test)

# naive is also known as persistence model.
naive_hat = zeros(len(y_hat), dtype = float32)
for i in range(0, len(y_hat)):
    # naive label is the label as horizon time steps before.
    # we have to consider to use only the fifth label here, too.
    naive_hat[i] = y[train_to + (i * test_step) - horizon]

# computing the mean squared errors of KNN and naive prediction.
mse_y_hat=mean_squared_error(y_test, y_hat)
mse_naive_hat=mean_squared_error(y_test, naive_hat)
print "MSE y_hat (KNN-Regressor): ", mse_y_hat
print "MSE naive_hat (Persistence): ", mse_naive_hat

    figure = plt.figure(figsize=(8, 5))
    time = range(0, len(y_hat))
    plt.plot(time, y_test, label="Measurement")
    plt.plot(time, naive_hat, label="Naive Label")
    plt.plot(time, y_hat, label="KNN Label")
    plt.xlim([2400, 2800])
    plt.ylim([-5, 50])
    plt.xlabel("Time Steps")
    plt.ylabel("Wind Power [MW]")


Fork me on GitHub