java - 使用导出的 sklearn 模型时 Android Studio 中的 "Code too large"

Question

目前我正在尝试将模型从 sklearn 导出到 Android。为此，我使用库 sklearn-porter ( sklearn-porter )。

这会从经过训练的模型生成一个 Java 类，如下所示:

class DecisionTreeClassifier {

   public static int predict(double[] features) {
        int[] classes = new int[2];

        if (features[350] <= 0.5156863033771515) {
            if (features[568] <= 0.0019607844296842813) {
                if (features[430] <= 0.0019607844296842813) {
                    if (features[405] <= 0.009803921915590763) {
...
}

此文件的大小约为 1 MB，因此在 Android Studio 中出现“Code too large”错误。

这个问题有解决方案吗？

Answer 1

当您使用 export_data=True 执行搬运工时，您还会得到一个小的 java 类，它从 .json 文件中读取 DecisionTree 参数:

porter = Porter(clf, language='java')
output = porter.export(export_data=True)
print(output)