LLMS聚合算法的加权融合

一、基础聚合:多模型输出投票/加权融合

1. 多数投票聚合器

java

import java.util.*;
import java.util.stream.Collectors;

public class VoteAggregator {
    
    /**
     * 多数投票聚合
     * @param results 多个模型输出的结果列表
     * @return 投票最多的结果
     */
    public static String majorityVote(List<String> results) {
        Map<String, Long> voteCount = results.stream()
            .collect(Collectors.groupingBy(
                result -> result, 
                Collectors.counting()
            ));
        
        return voteCount.entrySet().stream()
            .max(Map.Entry.comparingByValue())
            .map(Map.Entry::getKey)
            .orElse(null);
    }
    
    /**
     * 加权投票聚合
     * @param results 模型输出结果列表
     * @param weights 对应的权重列表(权重之和应为1)
     * @return 加权投票得分最高的结果
     */
    public static String weightedVote(List<String> results, List<Double> weights) {
        Map<String, Double> scoreMap = new HashMap<>();
        
        for (int i = 0; i < results.size(); i++) {
            String result = results.get(i);
            Double weight = weights.get(i);
            scoreMap.merge(result, weight, Double::sum);
        }
        
        return scoreMap.entrySet().stream()
            .max(Map.Entry.comparingByValue())
            .map(Map.Entry::getKey)
            .orElse(null);
    }
    
    // 使用示例
    public static void main(String[] args) {
        List<String> llmOutputs = Arrays.asList("positive", "neutral", "positive");
        System.out.println(majorityVote(llmOutputs)); // 输出: positive
        
        List<Double> weights = Arrays.asList(0.5, 0.3, 0.2);
        System.out.println(weightedVote(llmOutputs, weights)); // 输出: positive
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值