求夏普利值代码

我是嘉心糖

于 2023-06-12 12:58:43 发布

阅读量768

点赞数 1

分类专栏：联邦学习数学投资文章标签： java 算法开发语言联邦学习

本文链接：https://blog.csdn.net/qq_43445553/article/details/131166468

版权

数学同时被 3 个专栏收录

9 篇文章 0 订阅

订阅专栏

联邦学习

3 篇文章 1 订阅

订阅专栏

投资

2 篇文章 0 订阅

订阅专栏

该篇文章提供了一个Java程序，用于计算夏普利值。程序通过递归找到包含特定元素的所有子集，然后根据这些子集计算夏普利值，该值用于衡量个体在合作中的贡献。示例中展示了如何处理不同角色（如老板、工程师和打工仔）的贡献评估。

摘要由CSDN通过智能技术生成

本文提供了求夏普利值的代码，需要算法的地方只有分割子集。

import java.util.*;

public class Shapley {

    /**
     * 得到包含这个元素的全部子集
     * @param set
     * @param target
     * @return
     */
    public static Set<Set<String>> findSubsets(List<String> set, String target) {
        Set<Set<String>> subsets = new HashSet<>();
        findSubsets(set, target, 0, new HashSet<>(), subsets);
        return subsets;
    }

    private static void findSubsets(List<String> set, String target, int index, Set<String> currentSubset, Set<Set<String>> subsets) {
        if (index == set.size()) {
            if (currentSubset.contains(target)) {
                subsets.add(new HashSet<>(currentSubset));
            }
            return;
        }

        // Include the current element
        currentSubset.add(set.get(index));
        findSubsets(set, target, index + 1, currentSubset, subsets);

        // Exclude the current element
        currentSubset.remove(set.get(index));
        findSubsets(set, target, index + 1, currentSubset, subsets);
    }

    /**
     *
     * @param map 集合和钱之间的映射
     * @param target 需要求夏普利值的函数
     * @param set 包含全部元素的集合
     * @return 夏普利值
     */
    public static double getShapley(HashMap<Set<String>,Integer> map,String target,List<String> set){
        Set<Set<String>> subSetI = findSubsets(set,target);
        double sum = 0;
        for (Set<String> subSet:
                subSetI ) {
            double w = calW(new HashSet<>(set),subSet);
            double value1 = getValue(map,subSet);
            Set<String> removeSet = new HashSet<>(subSet);
            removeSet.remove(target);
            double value2 = getValue(map,removeSet);
            double add = w*(value1-value2);
            sum += add;
        }
        return sum;
    }
    public static void main(String[] args) {
        HashMap<Set<String>,Integer> map = new HashMap(){};
        map.put(new HashSet<>(Arrays.asList(new String[]{"老板"})),1);
        map.put(new HashSet<>(Arrays.asList(new String[]{"工程师"})),1);
        map.put(new HashSet<>(Arrays.asList(new String[]{"打工仔"})),0);
        map.put(new HashSet<>(Arrays.asList(new String[]{"老板","工程师"})),3);
        map.put(new HashSet<>(Arrays.asList(new String[]{"老板","打工仔"})),2);
        map.put(new HashSet<>(Arrays.asList(new String[]{"老板","工程师","打工仔"})),5);
        map.put(new HashSet<>(Arrays.asList(new String[]{"工程师","打工仔"})),2);

        String boss = "老板";
        String engineer = "工程师";
        String worker = "打工仔";
        String target = boss;
        List<String> set = new ArrayList<>(3);
        set.add(boss);
        set.add(engineer);
        set.add(worker);

        System.out.println("老板的夏普利值："+getShapley(map,boss,set));
        System.out.println("工程师的夏普利值："+getShapley(map,engineer,set));
        System.out.println("打工仔的夏普利值："+getShapley(map,worker,set));

    }

    /**
     *
     * @param map 集合和其赚到的钱的映射
     * @param s 需要求赚的钱的集合
     * @return 赚到的钱
     */
    public static double getValue(HashMap<Set<String>,Integer> map,Set<String> s){
        if(s.size()==0)return 0;
        for (Set<String> iterSet:
             map.keySet()) {
            if(iterSet.equals(s)){
                return map.get(iterSet);
            }
        }
        throw new RuntimeException("this set does not contain the element");
    }

    /**
     *
     * @param originSet 原始的集合，里面保管所有的元素
     * @param s 需要求权重的集合
     * @return 权重值
     */
    public static double calW(Set<String> originSet,Set<String> s){
        int sSize = s.size();
        int oSize = originSet.size();
        return (double) factorial(sSize-1)*factorial(oSize-sSize)/factorial(oSize);
    }

    /**
     * 求阶乘
     * @param n
     * @return
     */
    static int factorial(int n) {
        if (n == 0)
            return 1;
        else
            return (n * factorial(n - 1));
    }
}