def train()

in src/train.py [0:0]


def train():
    html_files_path = '/Users/Vbaungally/Downloads/autofill_html_dataset'
    artifact_path = 'artifacts' # saves models to this path
    ds = generate_dataset(html_files_path, samples=20000)
    print(ds)
    model = AutofillModel()
    model.load_data(ds)
    model.train(artifact_path)