CRF的多线程向前/向后

import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;

public class CRFMultithreaded {
    public static void main(String[] args) throws Exception {
        // 初始化CRF模型和输入数据
        CRFModel crfModel = new CRFModel();
        Sequence inputSequence = new Sequence();

        // 创建线程池
        ExecutorService executor = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors());

        // 分割输入数据并提交任务
        int chunkSize = inputSequence.size() / Runtime.getRuntime().availableProcessors();
        for (int i = 0; i < inputSequence.size(); i += chunkSize) {
            int end = Math.min(i + chunkSize, inputSequence.size());
            Sequence subSequence = inputSequence.subSequence(i, end);
            Future<Double> future = executor.submit(() -> crfModel.computeForwardBackward(subSequence));
            futures.add(future);
        }

        // 收集结果并合并
        double result = 0;
        for (Future<Double> future : futures) {
            result += future.get();
        }

        // 关闭线程池
        executor.shutdown();

        System.out.println("CRF计算结果: " + result);
    }
}