翻譯自:http://pythonhosted.org/coremltools/coremltools.utils.html?
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 可以輸出更多信息.
參考
例:
>>>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
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
保存路徑.
參考
例:
>>>coremltools.utils.save_spec(spec,'HousePricer.mlmodel')