"""
for n in N:
    for s in sigma:
        X3b_train,X3b_test,y3b_train,y3b_test = train_test_split(X2,y2,test_size= n, random_state = s)
        gr_test = []
        gr_train = []
        for i in range(1,21):
            fi_train = PolynomialFeatures(i).fit_transform(X3b_train.reshape(-1,1))
            w_train = matmul(pinv(fi_train), y3b_train)
            h_train = matmul(w_train,fi_train.transpose())

            fi_test = PolynomialFeatures(i).fit_transform(X3b_test.reshape(-1,1))
            h_test = matmul(w_train,fi_test.transpose())

            gr_test.append(np.log(mean_squared_error(y3b_test,h_test)))
            gr_train.append(np.log(mean_squared_error(y3b_train,h_train)))

        ax = fig.add_subplot(3,3,k+1)
        k+=1;
        plt.title('N = %d,   Šum = %d' % (n, s) )
        plt.plot(range(1,21),gr_train,range(1,21),gr_test)
        plt.legend(['train','test'],loc='best'),plt.grid()
"""