I use a decision tree model to do the prediction but when I tried to turn the .py file into an .onnx file, I got an error of 'DecisionTreeClassifier' object has no attribute 'graph'. I wonder is this means I have to submit a Neural network model to slove the challenge?
Created by Baiheng Chen bhchen @bhchen You can use any choice of model to train, as part of submission requirement you simply need to convert your trained model into onnx format. In your case simply convert your decision tree trained model into onnx format model.
sklearn has a package sklearn-onnx which convert trained model into onnx model. Below is the example for refernce.
https://onnx.ai/sklearn-onnx/
Drop files to upload
a question about truning the model into onnx file page is loading…