Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
636 changes: 305 additions & 331 deletions ai_lab_repo.py

Large diffs are not rendered by default.

360 changes: 247 additions & 113 deletions app.py

Large diffs are not rendered by default.

349 changes: 167 additions & 182 deletions inference.py

Large diffs are not rendered by default.

2 changes: 1 addition & 1 deletion mlesolver.py
Original file line number Diff line number Diff line change
Expand Up @@ -517,7 +517,7 @@ def phase_prompt(self,):
phase_str = (
"You are an ML engineer and you will be writing the code for a research project.\n"
"Your goal is to produce code that obtains final results for a set of research experiments. You should aim for simple code to collect all results, not complex code. You should integrate the provided literature review and the plan to make sure you are implementing everything outlined in the plan. The dataset code will be added to the beginning of your code always, so this does not need to be rewritten. Make sure you do not write functions, only loose code.\n"
"I would recommend writing smaller code so you do not run out of time but make sure to work on all points in the plan in the same code. You code should run every experiment outlined in the plan for a single code.\n",
"I would recommend writing smaller code so you do not run out of time but make sure to work on all points in the plan in the same code. You code should run every experiment outlined in the plan for a single code.\n"
"You cannot pip install new libraries, but many machine learning libraries already work. If you wish to use a language model in your code, please use the following:\nAnything you decide to print inside your code will be provided to you as input, and you will be able to see that part of the code. Using print statements is useful for figuring out what is wrong and understanding your code better."
)
return phase_str
Expand Down
9 changes: 6 additions & 3 deletions papersolver.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,9 +77,12 @@ def matches_command(self, cmd_str) -> bool:
def parse_command(self, *args) -> tuple:
sum_text = extract_prompt(args[0], "SUMMARY").split("\n")
full_text = extract_prompt(args[0], "FULL_TEXT").split("\n")
if len(sum_text) == 0 and len(full_text) == 0: return False, None
if len(sum_text) > 0: return True, ("SUMMARY", sum_text,)
if len(full_text) > 0: return True, ("FULL_TEXT", sum_text,)
if len(sum_text) == 0 and len(full_text) == 0:
return False, None
if len(sum_text) > 0 and sum_text[0] != "":
return True, ("SUMMARY", sum_text,)
if len(full_text) > 0 and full_text[0] != "":
return True, ("FULL_TEXT", full_text,)


"""
Expand Down
162 changes: 0 additions & 162 deletions readme/README-arabic.md

This file was deleted.

Loading