one hot java版

最新推荐文章于 2022-12-14 19:34:38 发布

喝酒当然老村长

最新推荐文章于 2022-12-14 19:34:38 发布

阅读量474

点赞数

分类专栏：日常笔记文章标签： java 机器学习

本文链接：https://blog.csdn.net/qq_45388979/article/details/123894190

版权

日常笔记专栏收录该内容

2 篇文章 0 订阅

订阅专栏

public static List<Integer> oneHot(JSONArray list, Double index) {

        int size = list.size();
        List<Integer> sumInt = new ArrayList<>();
        for (int i = 0; i < size + 1; i++) {
            sumInt.add(0);
        }
        for (int i = 0; i < list.size(); i++) {
            if (index <= list.getDouble(i)) {
                sumInt.set(i, 1);
                break;
            } else if (list.size() - 1 == i) {
                sumInt.set(i + 1, 1);
                break;
            }
        }
        return sumInt;
    }