code: Corrections and cleaning up
This commit is contained in:
		@ -27,7 +27,7 @@ def main(args):
 | 
			
		||||
  # The landmarks percentages
 | 
			
		||||
  lmdks_pct = [0, 20, 40, 60, 80, 100]
 | 
			
		||||
  # The privacy budget
 | 
			
		||||
  epsilon = 1.0
 | 
			
		||||
  epsilon = .1
 | 
			
		||||
 | 
			
		||||
  # Number of methods
 | 
			
		||||
  n = 3
 | 
			
		||||
@ -68,32 +68,26 @@ def main(args):
 | 
			
		||||
      # Skip
 | 
			
		||||
      rls_data_s, bgts_s = lmdk_bgt.skip_cont(seq, lmdks, epsilon)
 | 
			
		||||
      # lmdk_bgt.validate_bgts(seq, lmdks, epsilon, bgts_s)
 | 
			
		||||
      mae_s[i] += lmdk_bgt.mae_cont(rls_data_s)/args.iter
 | 
			
		||||
      mae_s[i] += (lmdk_bgt.mae_cont(rls_data_s)/args.iter)*100
 | 
			
		||||
 | 
			
		||||
      # Uniform
 | 
			
		||||
      rls_data_u, bgts_u = lmdk_bgt.uniform_cont(seq, lmdks, epsilon)
 | 
			
		||||
      # lmdk_bgt.validate_bgts(seq, lmdks, epsilon, bgts_u)
 | 
			
		||||
      mae_u[i] += lmdk_bgt.mae_cont(rls_data_u)/args.iter
 | 
			
		||||
      mae_u[i] += (lmdk_bgt.mae_cont(rls_data_u)/args.iter)*100
 | 
			
		||||
 | 
			
		||||
      # Adaptive
 | 
			
		||||
      rls_data_a, _, _ = lmdk_bgt.adaptive_cont(seq, lmdks, epsilon, .5, .5)
 | 
			
		||||
      mae_a[i] += lmdk_bgt.mae_cont(rls_data_a)/args.iter
 | 
			
		||||
      mae_a[i] += (lmdk_bgt.mae_cont(rls_data_a)/args.iter)*100
 | 
			
		||||
 | 
			
		||||
      # Calculate once
 | 
			
		||||
      if i == 0:
 | 
			
		||||
        # Event
 | 
			
		||||
        rls_data_evt, _ = lmdk_bgt.uniform_cont(seq, lmdk_lib.find_lmdks_cont(lmdk_data, seq, uid, 0), epsilon)
 | 
			
		||||
        mae_evt += lmdk_bgt.mae_cont(rls_data_evt)/args.iter
 | 
			
		||||
        mae_evt += (lmdk_bgt.mae_cont(rls_data_evt)/args.iter)*100
 | 
			
		||||
        # User
 | 
			
		||||
        rls_data_usr, _ = lmdk_bgt.uniform_cont(seq, lmdk_lib.find_lmdks_cont(lmdk_data, seq, uid, 100), epsilon)
 | 
			
		||||
        mae_usr += lmdk_bgt.mae_cont(rls_data_usr)/args.iter
 | 
			
		||||
 | 
			
		||||
  mae_u *= 100
 | 
			
		||||
  mae_s *= 100
 | 
			
		||||
  mae_a *= 100
 | 
			
		||||
  mae_evt *= 100
 | 
			
		||||
  mae_usr *= 100
 | 
			
		||||
 | 
			
		||||
        mae_usr += (lmdk_bgt.mae_cont(rls_data_usr)/args.iter)*100
 | 
			
		||||
  exit()
 | 
			
		||||
  plt.axhline(
 | 
			
		||||
    y = mae_evt,
 | 
			
		||||
    color = '#212121',
 | 
			
		||||
@ -172,7 +166,7 @@ if __name__ == '__main__':
 | 
			
		||||
    main(parse_args())
 | 
			
		||||
    end_time = time.time()
 | 
			
		||||
    print('##############################')
 | 
			
		||||
    print('Time   : %.4fs' % (end_time - start_time))
 | 
			
		||||
    print('Time elapsed: %s' % (time.strftime('%H:%M:%S', time.gmtime(end_time - start_time))))
 | 
			
		||||
    print('##############################')
 | 
			
		||||
  except KeyboardInterrupt:
 | 
			
		||||
    print('Interrupted by user.')
 | 
			
		||||
 | 
			
		||||
		Reference in New Issue
	
	Block a user