tensorflow - 如何将保存的模型从sklearn转换为tensorflow/lite

Question

如果我想使用sklearn库实现分类器。有没有一种方法可以将模型另存为或将文件转换为已保存的tensorflow文件，以便稍后将其转换为tensorflow lite？

Answer 1

如果您在TensorFlow中复制架构，由于scikit-learn模型通常非常简单，这将非常容易，则可以将学习到的scikit-learn模型中的参数显式分配给TensorFlow层。
这是将逻辑回归转换为单个密集层的示例:

import tensorflow as tf
import numpy as np
from sklearn.linear_model import LogisticRegression

# some random data to train and test on
x = np.random.normal(size=(60, 21))
y = np.random.uniform(size=(60,)) > 0.5

# fit the sklearn model on the data
sklearn_model = LogisticRegression().fit(x, y)

# create a TF model with the same architecture
tf_model = tf.keras.models.Sequential()
tf_model.add(tf.keras.Input(shape=(21,)))
tf_model.add(tf.keras.layers.Dense(1))

# assign the parameters from sklearn to the TF model
tf_model.layers[0].weights[0].assign(sklearn_model.coef_.transpose())
tf_model.layers[0].bias.assign(sklearn_model.intercept_)

# verify the models do the same prediction
assert np.all((tf_model(x) > 0)[:, 0].numpy() == sklearn_model.predict(x))