机器学习

numpy:axis中的值代表数组的维度（a.max(axis=1/0)）

pandas:axis中的值代表数组的维度（df.dropna(axis = 1/0)删除有缺失数据的列/行）

机器学习模型

(1)

y：预测的值

X:特征值

from sklearn.tree import DecisionTreeRegressor

model=DecisionTreeRegressor(random_state=1)

model.fit(X,y)

predict(X)

(2)

MAE and 分割数据

from sklearn.metrics import mean_absolute_error
from sklearn.model_selection import train_test_split
train_X, val_X, train_y, val_y = train_test_split(X, y, random_state = 0)
melbourne_model = DecisionTreeRegressor()
melbourne_model.fit(train_X, train_y)
val_predictions = melbourne_model.predict(val_X)
print(mean_absolute_error(val_y, val_predictions))

 (3)

生成最优决策树

from sklearn.metrics import mean_absolute_error from sklearn.tree import DecisionTreeRegressor def get_mae(max_leaf_nodes, train_X, val_X, train_y, val_y): model = DecisionTreeRegressor(max_leaf_nodes=max_leaf_nodes, random_state=0) model.fit(train_X, train_y) preds_val = model.predict(val_X) mae = mean_absolute_error(val_y, preds_val) return(mae)

随机林

from sklearn.ensemble import RandomForestRegressor
from sklearn.metrics import mean_absolute_error

forest_model = RandomForestRegressor(random_state=1)
forest_model.fit(train_X, train_y)
melb_preds = forest_model.predict(val_X)
print(mean_absolute_error(val_y, melb_preds))