iOS CoreML 模型轉(zhuǎn)換工具coremltools(三)

翻譯自:http://pythonhosted.org/coremltools/coremltools.utils.html?

Utilities

coremltools.models.utils.evaluate_classifier(model,data,target='target',verbose=False)

評估 CoreML 分類器模型同時 同原始的模型做比較 (用于測試轉(zhuǎn)換后的正確性).

Parameters:filename: [str | MLModel]

模型文件或者 MLModel.

data: [str | Dataframe]

評估模型的測試數(shù)據(jù) (dataframe 或者 csv文件路徑).

target: str

目標(biāo)列名稱

verbose: bool

設(shè)置為 true 可以輸出更多信息.

參考

evaluate_regressor,evaluate_classifier_with_probabilities

例:

>>>metrics=coremltools.utils.evaluate_classifier(spec,'data_and_predictions.csv','target')>>>printmetrics{"samples": 10, num_errors: 0}

coremltools.models.utils.evaluate_classifier_with_probabilities(model,data,probabilities='probabilities',verbose=False)

評估 CoreML 分類器模型.

Parameters:filename: [str | Model]

模型文件或者 MLModel.

data: [str | Dataframe]

評估模型的測試數(shù)據(jù) (dataframe 或者 csv文件路徑).

probabilities: str

目標(biāo)列名稱

verbose: bool

設(shè)置為 true 可以輸出更多信息.

coremltools.models.utils.evaluate_regressor(model,data,target='target',verbose=False)

評估 CoreML 回歸模型同時 同原始的模型做比較 (用于測試轉(zhuǎn)換后的正確性)

Parameters:filename: [str | MLModel]

模型文件或者 MLModel.

data: [str | Dataframe]

評估模型的測試數(shù)據(jù) (dataframe 或者 csv文件路徑).

target: str

目標(biāo)列名稱.

verbose: bool

設(shè)置為 true 可以輸出更多信息.

參考

evaluate_classifier

例:

>>>metrics=coremltools.utils.evaluate_regressor(spec,'data_and_predictions.csv','target')>>>printmetrics{"samples": 10, "rmse": 0.0, max_error: 0.0}

coremltools.models.utils.evaluate_transformer(model,input_data,reference_output,verbose=False)

Evaluate a transformer specification for testing.

Parameters:spec: [str | MLModel]

模型文件或者 MLModel.

input_data: list[dict]

評估模型的測試數(shù)據(jù).

reference_output: list[dict]

模型的預(yù)期結(jié)果.

verbose: bool

設(shè)置為 true 可以輸出更多信息.

參考

evaluate_regressor,evaluate_classifier

例:

>>>input_data=[{'input_1':1,'input_2':2},{'input_1':3,'input_2':3}]>>>expected_output=[{'input_1':2.5,'input_2':2.0},{'input_1':1.3,'input_2':2.3}]>>>metrics=coremltools.utils.evaluate_transformer(scaler_spec,input_data,expected_output)

coremltools.models.utils.load_spec(filename)

從文件中加載protobuf 模型規(guī)范

Parameters:filename: str

能夠加載到protobuf文件的磁盤位置 (合法的文件路徑) .

Returns:model_spec: Model_pb

模型的Protobuf 表示

See also

save_spec

Examples

>>>spec=coremltools.utils.load_spec('HousePricer.mlmodel')

coremltools.models.utils.rename_feature(spec,current_name,new_name,rename_inputs=True,rename_outputs=True)

重命名規(guī)范中的特征名.

Parameters:spec: Model_pb

包含將要將要命名的特征的規(guī)范.

current_name: str

當(dāng)前特征名稱. 如果特征不存在, 此次調(diào)用不做任何操作.

new_name: str

修改后的特征名稱.

rename_inputs: bool

只修改輸入中的特征 (忽略輸出特征)

rename_outputs: bool

只修改輸出中的特征 (忽略輸入特征)

例:

# In-place rename of spec>>>coremltools.utils.rename_feature(spec,'old_feature','new_feature_name')

coremltools.models.utils.save_spec(spec,filename)

將 protobuf 模型說明保存到文件.

Parameters:spec: Model_pb

Protobuf 模型

filename: str

保存路徑.

參考

load_spec

例:

>>>coremltools.utils.save_spec(spec,'HousePricer.mlmodel')

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

  • Spring Cloud為開發(fā)人員提供了快速構(gòu)建分布式系統(tǒng)中一些常見模式的工具(例如配置管理,服務(wù)發(fā)現(xiàn),斷路器,智...
    卡卡羅2017閱讀 136,512評論 19 139
  • 早上來公司的路上,經(jīng)過一片林蔭道,有三兩片青黃相接的樹葉從樹上飄落,掉在我的腳前。 有些許微風(fēng)吹過,別...
    淼淼hua閱讀 628評論 9 8
  • 我記得那是2011年春運時候的事情。那時候我還在讀本科,屬于窮得叮當(dāng)響的時間。那一年是從西藏坐火車硬座到長沙。記不...
    斑斑的四喜丸子閱讀 322評論 0 4
  • 此刻的我,有滿身彩虹般可見的彩 卻又有著,滿心夜影般捉不到的黑 你若問我,從哪里來, 經(jīng)過了什么地方,要去向何處...
    自我文理閱讀 462評論 0 0

友情鏈接更多精彩內(nèi)容