浏览代码

bug fix with explore policy

Steve Nyemba 6 年之前
父节点
当前提交
30fe9cf9fc
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      risk/risk.py

+ 1 - 1
risk/risk.py

@@ -76,7 +76,7 @@ class deid :
         
         
         sample = args['sample'] if 'sample' in args else pd.DataFrame(self._df)
         sample = args['sample'] if 'sample' in args else pd.DataFrame(self._df)
         
         
-        k = sample.columns.size -1 if 'field_count' not in args else int(args['field_count'])
+        k = sample.columns.size if 'field_count' not in args else int(args['field_count']) + 1
         if 'id' in args :
         if 'id' in args :
             id = args['id']
             id = args['id']
             columns = list(set(sample.columns.tolist()) - set([id]))
             columns = list(set(sample.columns.tolist()) - set([id]))