Facebook
From Sloppy Plover, 7 Years ago, written in Plain Text.
Embed
Download Paste or View Raw
Hits: 274
  1. def design_matrix(x_train, M):
  2.     '''
  3.     :param x_train: ciag treningowy Nx1
  4.     :param M: stopien wielomianu 0,1,2,...
  5.     :return: funkcja wylicza Design Matrix Nx(M+1) dla wielomianu rzedu M
  6.     matrix = np.ones(shape=(x_train.shape[0], M+1))
  7.     for col in range(M + 1):
  8.         for row in range(matrix.shape[0]):
  9.             matrix[row][col] = pow(x_train[row][0], col)
  10.     '''
  11.     return np.array([[pow(x_train[row][0], col) for col in range(M+1)] for row in range(x_train.shape[0])])