github/home-llm
1
1
Fork 0
mirror of https://github.com/acon96/home-llm.git synced 2026-01-09 21:58:00 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix evaluate + add train notebook

Browse Source
This commit is contained in:
Alex O'Connell
2025-02-10 17:09:49 -05:00
parent fca80d0504
commit 2712f605a5
4 changed files with 411 additions and 289 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
data/generate_home_assistant_data.py
View File

@@ -1151,7 +1151,7 @@ def load_dataset_piles(language):
# TODO: answer questions about more than one thing in the state list at once
# TODO: add examples for rooms/groups of devices. i.e. "turn off all the lights in the kitchen"
# TODO: add time, weather, and calendar/reminders (next 3 events?)
def main():
def main(args=None):
parser = argparse.ArgumentParser(description="Generate the full dataset from the CSV piles")
parser.add_argument("--sample", action="store_true", help="Set this flag to enable generation of the train dataset.")
parser.add_argument("--test", action="store_true", help="Set this flag to enable generation of the train dataset.")
@@ -1171,7 +1171,7 @@ def main():
dataset_format_group.add_argument('--raw_corpus', action='store_const', const='raw', dest='format')
dataset_format_group.add_argument('--sharegpt', action='store_const', const='sharegpt', dest='format')
args = parser.parse_args()
args = parser.parse_args(args=args)
if not args.sample and not args.train and not args.test and not args.merge and not args.dpo:
parser.print_usage()