gpt4 book ai didi

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

转载 作者:塔克拉玛干 更新时间:2023-11-02 19:05:16 26 4
gpt4 key购买 nike

目前我正在尝试将模型从 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”错误。

这个问题有解决方案吗?

最佳答案

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

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

关于java - 使用导出的 sklearn 模型时 Android Studio 中的 "Code too large",我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/58289249/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com