Scikit-learn의 DecisionTreeClassifier 모델 사용 문의

한국어 (Korean) 일반
1

image
image980×993 39.5 KB

해당 구조의 DecisionTreeClassifier를 학습시켜 사용하고자 하는데
Furiosa SDK에선 optimize_model을 실행할 경우 ai.onnx.ml 연산자를 지원하지 않는 것으로 보입니다.

  1. SDK에서 지원하는 연산자 목록을 알 수 있을까요?

  2. Calibrator 생성 과정에서 해당 모델을 onnxruntime에서 양자화하는 과정 중
    SafeIntOnOverflow() Integer overflow에러가 발생합니다.

    캡처2
    캡처21223×105 6.63 KB

해당 에러가 발생하는 지점을 찾을 수가 없게 오류 메세지가 떠서
이 에러가 발생하는 부분을 알 수 있나요?

2

안녕하세요, FuriosaAI 김종욱입니다. 질문주신 내용에 대하여 답변을 드리면,

  1. 저희 SDK에서 지원하는 연산자 목록은 SDK 문서에서 확인 가능하십니다.
  2. 해당 에러가 발생하는 지점에 대해서는 모델을 보아야 확인이 가능할 것으로 보입니다.

현재 SDK에서 지원하는 ONNX opset version이 12~13에 대하여 지원을 하고 있어 opset version을 13으로 하여 onnx 모델을 만드신 이후 테스트 진행해주시면 정확한 문제를 파악할 수 있을 것 같습니다!

1 Like