def test_wp_save_load_train():
happy_wp = HappyWordPrediction('', 'distilroberta-base')
output_path = "data/wp-train.json"
data_path = "../data/wp/train-eval.txt"
args = ARGS_WP_TRAIN
args["line_by_line"] = True
run_save_load(happy_wp, output_path, args, data_path, "train")
def test_tc_save_load_eval():
happy_wp = HappyTextClassification()
output_path = "data/tc-train.json"
data_path = "../data/tc/train-eval.csv"
run_save_load(happy_wp, output_path, ARGS_TC_EVAL, data_path, "eval")
def test_gen_save_load_eval():
happy_gen = HappyGeneration()
output_path = "data/wp-eval.txt"
data_path = "../data/gen/train-eval.txt"
run_save_load(happy_gen, output_path, ARGS_GEN_EVAl, data_path, "eval")
def test_gen_save_load_train():
happy_gen = HappyGeneration()
output_path = "data/gen-train.txt"
data_path = "../data/gen/train-eval.txt"
run_save_load(happy_gen, output_path, ARGS_GEN_TRAIN, data_path, "train")