问题描述
我正在尝试将sklearn SVC模型转换/存储为.onnx文件,并且遇到了我不理解的运行时错误。通过sklearn随机森林分类器和sklearn k-NN分类器,我已经能够有效地使用相同的代码而不会出现错误。我收到一个我不理解的奇怪的onnx运行时错误。感谢您提供有关此错误的任何帮助。
我首先在下面发布了运行文件svm_time.py的输出,然后在下面显示了svm_time.py文件中包含的代码。
谢谢。
df %>%
mutate(x2 = case_when(a == 3 ~ 0.05 + x))
python3 svm_time.py
'train_model' 4809.58 ms
train score is: 0.8765468473777254
val Accuracy is: 0.7037037037037037
Traceback (most recent call last):
File "svm_time.py",line 97,in <module>
main()
File "svm_time.py",line 91,in main
onx = convert_sklearn(clf,initial_types=initial_type)
File "/home/matt/anaconda3/envs/venv/lib/python3.7/site-packages/skl2onnx/convert.py",line 154,in convert_sklearn
dtype=dtype,options=options)
File "/home/matt/anaconda3/envs/venv/lib/python3.7/site-packages/skl2onnx/common/_topology.py",line 1054,in convert_topology
conv(scope,operator,container)
File "/home/matt/anaconda3/envs/venv/lib/python3.7/site-packages/skl2onnx/common/_registration.py",line 29,in __call__
return self._fct(*args)
File "/home/matt/anaconda3/envs/venv/lib/python3.7/site-packages/skl2onnx/operator_converters/support_vector_machines.py",line 221,in convert_sklearn_svm_classifier
"Classes different from first n integers are not supported "
RuntimeError: Classes different from first n integers are not supported in SVC converter.
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)