這是一個定製的模型基於sklearn模型“GradientBoostingClassifier”:
類CustomizedGradientBoostingClassifier (sklearn.ensemble.GradientBoostingClassifier): def __init__(自我,random_state):超級(). __init__ (random_state = random_state) def適合(自我,X, y):超級()。fit (X, y) def predict_proba(自我,X_test):超級()返回.predict_proba (X_test) def預測(自我,X): #做定製的任務(例如下一個RPC調用日誌輸入和輸出)#例如,您也可以返回不僅預測結果,而且輸入返回(super () .predict (X), X)
你可以注冊模型像往常一樣。當您調用REST端點,它一些自定義的預測()函數,並返回不僅預測結果,而且輸入。