第七色在线视频,2021少妇久久久久久久久久,亚洲欧洲精品成人久久av18,亚洲国产精品特色大片观看完整版,孙宇晨将参加特朗普的晚宴

為了賬號(hào)安全,請(qǐng)及時(shí)綁定郵箱和手機(jī)立即綁定
已解決430363個(gè)問題,去搜搜看,總會(huì)有你想問的

在 for 循環(huán)中保存 scikit-learn 模型

在 for 循環(huán)中保存 scikit-learn 模型

暮色呼如 2021-12-09 10:48:40
我正在使用 scikit-learn 運(yùn)行一堆模型來解決分類問題。這是應(yīng)該完成所有運(yùn)行的代碼:for model_name, classifier, param_grid, cv, cv_name in tqdm(zip(model_names, classifiers, param_grids, cvs, cv_names)):    pipeline = Pipeline(steps=[('preprocessor', preprocessor),                     ('classifier', classifier)])     train_and_score_model(model_name, pipeline, param_grid, cv=cv)我的問題是,如何保留train_and_score_model函數(shù)的輸出?它返回一個(gè) cv 對(duì)象,即一個(gè)模型。我試圖做但我認(rèn)為不正確的是創(chuàng)建一個(gè)列表cv_names = ['dm_cv', 'lr_cv', 'knn_cv', 'svm_cv', 'dt_cv', 'rf_cv', 'nb_cv']并將每個(gè)列表設(shè)置為 for 循環(huán)運(yùn)行。那是cv_namefor 循環(huán)頭中的迭代器。我不認(rèn)為那是對(duì)的,因?yàn)槲也粫?huì)設(shè)置字符串而不是變量嗎?就像,我真正應(yīng)該擁有的是cv_names = [dm_cv, lr_cv, knn_cv, svm_cv, dt_cv, rf_cv, nb_cv],但我不認(rèn)為我可以擁有這樣的列表。我想到的另一種方法是將每個(gè)模型保存在字典中,其中的鍵是我上面概述的列表中的元素。我不知道我是否可以將模型作為字典值。這是我目前在 for 循環(huán)中運(yùn)行的笨重、重復(fù)的代碼:pipeline = Pipeline(steps=[('preprocessor', preprocessor),                 ('classifier', classifier_dm)])dm_cv = train_and_score_model('Dummy Model', pipeline, param_grid_dm)pipeline = Pipeline(steps=[('preprocessor', preprocessor),                     ('classifier', classifier_lr)])lr_cv = train_and_score_model('Logistic Regression', pipeline, param_grid_lr)pipeline = Pipeline(steps=[('preprocessor', preprocessor),                     ('classifier', classifier_knn)])knn_cv = train_and_score_model('K Nearest Neighbors', pipeline, param_grid_knn)pipeline = Pipeline(steps=[('preprocessor', preprocessor),                     ('classifier', classifier_svm)])svm_cv = train_and_score_model('Support Vector Machine', pipeline, param_grid_svm)pipeline = Pipeline(steps=[('preprocessor', preprocessor),                     ('classifier', classifier_dt)])dt_cv = train_and_score_model('Decision Tree', pipeline, param_grid_dt)
查看完整描述

1 回答

?
守候你守候我

TA貢獻(xiàn)1802條經(jīng)驗(yàn) 獲得超10個(gè)贊

您可以創(chuàng)建一個(gè)字典,其中包含分類器名稱及其信息(即對(duì)象和參數(shù)網(wǎng)格)的映射:


models_list = {'Logistic Regression': (classifier_lr, param_grid_lr),

               'K Nearest Neighbours': (classifier_knn, param_grid_knn)}

遍歷字典中的每個(gè)鍵值對(duì)并構(gòu)建您的管道:


model_cvs = {}

for model_name, model_info in models_list.items():

    pipeline = Pipeline(steps=[('preprocessor', preprocessor),

                       ('classifier', model_info[0])])

    model_cvs[model_name] = train_and_score_model(model_name, pipeline, model_info[1])


查看完整回答
反對(duì) 回復(fù) 2021-12-09
  • 1 回答
  • 0 關(guān)注
  • 192 瀏覽
慕課專欄
更多

添加回答

舉報(bào)

0/150
提交
取消
微信客服

購課補(bǔ)貼
聯(lián)系客服咨詢優(yōu)惠詳情

幫助反饋 APP下載

慕課網(wǎng)APP
您的移動(dòng)學(xué)習(xí)伙伴

公眾號(hào)

掃描二維碼
關(guān)注慕課網(wǎng)微信公眾號(hào)