Python數(shù)據(jù)科學(xué):神經(jīng)網(wǎng)絡(luò)
# 極值標(biāo)準(zhǔn)化處理
scaler = MinMaxScaler()
scaler.fit(train_data)
scaled_train_data = scaler.transform(train_data)
scaler_test_data = scaler.transform(test_data)
建立多層感知器模型。
# 設(shè)置多層感知器對應(yīng)的模型
mlp = MLPClassifier(hidden_layer_sizes=(10,), activation='logistic', alpha=0.1, max_iter=1000)
# 對訓(xùn)練集進(jìn)行模型訓(xùn)練
mlp.fit(scaled_train_data, train_target)
# 輸出神經(jīng)網(wǎng)絡(luò)模型信息
print(mlp)
輸出模型信息如下。
接下來使用經(jīng)過訓(xùn)練集訓(xùn)練的模型,對訓(xùn)練集及測試集進(jìn)行預(yù)測。
# 使用模型進(jìn)行預(yù)測
train_predict = mlp.predict(scaled_train_data)
test_predict = mlp.predict(scaler_test_data)
輸出預(yù)測概率,用戶流失的概率。
# 輸出模型預(yù)測概率(為1的情況)
train_proba = mlp.predict_proba(scaled_train_data)[:, 1]
test_proba = mlp.predict_proba(scaler_test_data)[:, 1]
對模型進(jìn)行評估,輸出評估數(shù)據(jù)。
# 根據(jù)預(yù)測信息輸出模型評估結(jié)果
print(metrics.confusion_matrix(test_target, test_predict, labels=[0, 1]))
print(metrics.classification_report(test_target, test_predict))
輸出如下。
模型對流失用戶的f1-score(精確率和召回率的調(diào)和平均數(shù))值為0.81,效果不錯。
此外對流失用戶的靈敏度recall為0.83,模型能識別出83%的流失用戶,說明模型識別流失用戶的能力還可以。
輸出模型預(yù)測的平均準(zhǔn)確度。
# 使用指定數(shù)據(jù)集輸出模型預(yù)測的平均準(zhǔn)確度
print(mlp.score(scaler_test_data, test_target))
# 輸出值為0.8282828282828283
平均準(zhǔn)確度值為0.8282。
計算模型的ROC下面積。
# 繪制ROC曲線
fpr_test, tpr_test, th_test = metrics.roc_curve(test_target, test_proba)
fpr_train, tpr_train, th_train = metrics.roc_curve(train_target, train_proba)
plt.figure(figsize=[3, 3])
plt.plot(fpr_test, tpr_test, 'b--')
plt.plot(fpr_train, tpr_train, 'r-')
plt.title('ROC curve')
plt.show()
# 計算AUC值
print(metrics.roc_auc_score(test_target, test_proba))
# 輸出值為0.9149632415075206
ROC曲線圖如下。
訓(xùn)練集和測試集的曲線很接近,沒有過擬合現(xiàn)象。
AUC值為0.9149,說明模型效果非常好。
對模型進(jìn)行最優(yōu)參數(shù)搜索,并且對最優(yōu)參數(shù)下的模型進(jìn)行訓(xùn)練。

發(fā)表評論
請輸入評論內(nèi)容...
請輸入評論/評論長度6~500個字
最新活動更多
-
7月8日立即報名>> 【在線會議】英飛凌新一代智能照明方案賦能綠色建筑與工業(yè)互聯(lián)
-
7月22-29日立即報名>> 【線下論壇】第三屆安富利汽車生態(tài)圈峰會
-
7.30-8.1火熱報名中>> 全數(shù)會2025(第六屆)機(jī)器人及智能工廠展
-
7月31日免費(fèi)預(yù)約>> OFweek 2025具身智能機(jī)器人產(chǎn)業(yè)技術(shù)創(chuàng)新應(yīng)用論壇
-
免費(fèi)參會立即報名>> 7月30日- 8月1日 2025全數(shù)會工業(yè)芯片與傳感儀表展
-
即日-2025.8.1立即下載>> 《2024智能制造產(chǎn)業(yè)高端化、智能化、綠色化發(fā)展藍(lán)皮書》
推薦專題