-
Notifications
You must be signed in to change notification settings - Fork 6
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Approximate selection #107
Conversation
@ddkang Could you please review this PR? |
@ddkang Could you please review this PR? |
aidb/engine/approx_select_engine.py
Outdated
dataset = self.get_sampled_proxy_blob(proxy_score_for_all_blobs) | ||
|
||
# This is used for parallel test | ||
seed = (mp.current_process().pid * np.random.randint(100000, size=1)[0]) % (2**32 - 1) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can you pass in the seed as an argument to the base engine? I don't think it's good to set here
aidb/engine/approx_select_engine.py
Outdated
|
||
logger.info(f'num_samples: {len(additional_samples)}') | ||
|
||
additional_satisfied_sampled_results, additional_all_sampled_results = await self.get_inference_results( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Don't we also want to take the positive records from the pilot sample?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Yes, we have taken the positive records from the pilot sample. The variable names are ambiguous, I will change them.
aidb/aidb/engine/approx_select_engine.py
Lines 212 to 215 in 4c67799
R1 = sorted_satisfied_sampled_results.index | |
R2 = dataset[dataset[PROXY_SCORE] >= tau_modified].index | |
additional_samples = list(set(R1).union(set(R2))) |
@ddkang Could you please review the new commit? |
https://github.com/ddkang/aidb/wiki/Apporx-selection-%E2%80%90-Design-Document