c2ba7a00469d0394a4a8073c355cfb884d788cc9,ch12/train_model.py,,,#,18

Before Change

    saves_path = os.path.join(SAVES_DIR, args.name)
    os.makedirs(saves_path, exist_ok=True)

    dialogues = subtitles.read_file(DATA_FILE, dialog_seconds=5)
    dial_dict = subtitles.dialogues_dict(dialogues)
    log.info("Loaded %d dialogues with %d phrases and %d uniq words",
             len(dialogues), sum(map(len, dialogues)), len(dial_dict))
    emb_dict, emb = data.read_embeddings(dial_dict)

After Change

    os.makedirs(saves_path, exist_ok=True)

    if args.category is not None:
        data_path = os.path.join(DATA_DIR, args.category)
    else:
        data_path = DATA_DIR
    dialogues = subtitles.read_dir(data_path)
    if not dialogues:
        log.error("No data found in %s!", data_path)
        sys.exit()

    dial_dict = subtitles.dialogues_dict(dialogues)
    log.info("Loaded %d dialogues with %d phrases and %d uniq words",
             len(dialogues), sum(map(len, dialogues)), len(dial_dict))
    emb_dict, emb = data.read_embeddings(dial_dict)

Instances