adding the strict flag

2019-03-08 18:02:06 -06:00 · 2019-03-08 18:02:06 -06:00 · 084e7d0fe8
parent e33223f6b2
commit 084e7d0fe8
1 changed files with 5 additions and 3 deletions
--- a/risk/risk.py
+++ b/risk/risk.py
@ -85,8 +85,10 @@ class deid :
        o = pd.DataFrame()
        
        for i in np.arange(RUNS):
+            if 'strict' not in args :
                n = np.random.randint(2,k)
-            
+            else:
+                n = args['field_count']
            cols = np.random.choice(columns,n,replace=False).tolist()            
            params = {'sample':sample,'cols':cols}
            if pop is not None :