How to convert KNN Scikit-learn Model in python to tensorflow lite model?

Question

This is my sample code for a KNN classifier with accuracy over 90%,

sc_X = StandardScaler()

 X_train = sc_X.fit_transform(X_train)

 X_test = sc_X.transform(X_test)

 k=10
 classifier=KNeighborsClassifier(n_neighbors=k)

 classifier.fit(X_train,y_train)

 y_pred=classifier.predict(X_test)

 acc=accuracy_score(y_test, y_pred)

print("For K=",k,"-->Accuracy is:",acc) 


Am trying to convert the above listed model to a tensor flow lite model using this,

converter = lite.TFLiteConverter.from_keras_model(classifier)

tfmodel = converter.convert()

open('trained_model.tflite', 'wb').write(tfmodel)

But i am getting this error,

'KNeighborsClassifier' object has no attribute 'call'

Is there anyway to convert a trained knn model in python to tflite model?

You can't convert a sklearn model to tensorflow or save it as a keras model. — Freddy Daniel
– Freddy Daniel, Commented Jul 29, 2021 at 17:09
Is there any simple implementation of knn model in keras or any other type, which is also accepted by tensor flow conversion, am trying to deploy a model in an android app (using android studio) — MoonLight
– MoonLight, Commented Jul 30, 2021 at 10:04

Sachin Joglekar · Accepted Answer · 2021-07-29 16:41:30Z

1

Looks like KNeighborsClassifier is part of the sklearn library. lite.TFLiteConverter.from_keras_model supports keras models, not sklearn models. You need to build & train a Keras classifier.

answered Jul 29, 2021 at 16:41

Sachin Joglekar

6964 silver badges5 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

MoonLight Over a year ago

Is there a simple implementation in keras for classification models just like with sklearn libraries ?

Sachin Joglekar Over a year ago

Yup, the codelab I linked in the answer shows a simple Keras classifier training, and you can customize it from there :-)

Collectives™ on Stack Overflow

How to convert KNN Scikit-learn Model in python to tensorflow lite model?

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related