本文主要向大家介绍了机器学习入门之基于python机器学习人脸自动补全,通过具体的内容向大家展现,希望对大家学习机器学习入门有所帮助。
import numpy as np
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression,Ridge,Lasso
from sklearn.neighbors import KNeighborsRegressor
from sklearn.ensemble import ExtraTreesRegressor
from sklearn.datasets import fetch_olivetti_faces
faces=fetch_olivetti_faces()
data=faces[‘data‘]
target=faces[‘target‘]
#data.shape
#人脸补全
#人脸数据一分为二,上半部分作为数据,下半部分作为target
face_up,face_down=data[:,:2048],data[:,2048:]
from sklearn.model_selection import train_test_split
x_train,x_test,y_train,y_test=train_test_split(face_up,face_down,test_size=0.1)
#5个算法分别识别
estimators={‘knn‘:KNeighborsRegressor(),
‘LinearRe‘:LinearRegression(),
‘Ridge‘:Ridge(alpha=0.1),
‘