typo

Author: jqu-striveworks

valor-sandbox/valor_sandbox.py

@@ -4,15 +4,15 @@
 from valor_implementation import _calculate_pr_curves
 from optimized_implementation import _calculate_pr_curves_optimized
 
-n = 1000
+n = 10000000
 n_class = 3
 predictions_per_datum = 3
 
 groundtruth_df = generate_groundtruth(n, n_class)
 prediction_df = generate_predictions(n, n_class, 3)
 
 print(f"Groundtruth Dataframe Memory Usage: {groundtruth_df.memory_usage(index=True).sum()}")
-print(f"Prediction Dataframe Memory Usage: {groundtruth_df.memory_usage(index=True).sum()}")
+print(f"Prediction Dataframe Memory Usage: {prediction_df.memory_usage(index=True).sum()}")
 
 ## Randomly drop some samples.
 ## groundtruth_df = groundtruth_df.sample(frac=0.9)
@@ -28,7 +28,6 @@
 end = time.time()
 print(f"Valor completed {n} in {end - start}")
 
-
 start = time.time()
 fast_df = _calculate_pr_curves_optimized(
     groundtruth_df=groundtruth_df,