micromlgen icon indicating copy to clipboard operation
micromlgen copied to clipboard

Can we use micromlgen with xgboost itself - without XGBClassifier

Open karimkhanvi opened this issue 3 years ago • 0 comments

micromlgen works well with XGBClassifier. Which is imported from xgboost.

from xgboost import XGBClassifier

But in my program, I am using xgboost without any importa like this

import xgboost as xgb
model = xgb.cv(param,dtrain,num_boost_round=420,nfold=10,stratified=True,verbose_eval=20)
model = xgb.train(param,dtrain,420,[(dtrain,'train')],verbose_eval=20)

I tried using port on model, which gives me error

Traceback (most recent call last):
  File "model_microml_training.py", line 161, in <module>
    predict(sys.argv[1])
  File "model_microml_training.py", line 151, in predict
    print(port(model))
  File "/home/admin/dawid_venv_xgboost_1.1.0/lib/python3.5/site-packages/micromlgen/micromlgen.py", line 45, in port
    raise TypeError('clf MUST be one of %s' % ', '.join(platforms.ALLOWED_CLASSIFIERS))
TypeError: clf MUST be one of SVC, OneClassSVC, RVC, SEFR, DecisionTree, RandomForest, GaussianNB, LogisticRegression, PCA, PrincipalFFT, LinearRegression, XGBClassifier

karimkhanvi avatar May 31 '21 15:05 karimkhanvi