FinGPT
FinGPT copied to clipboard
执行finetune.sh 之后,效果不理想
目前我爬取了5只股票(东方财富的资讯),大约有4w条数据,执行finetune.sh 日志如下图
执行脚本参数
python -u finetune.py
--dataset_path "/data/program/data/dataset6/jsonl/dataset_title_train_and_valid"
--lora_rank 8
--per_device_train_batch_size 16
--per_device_eval_batch_size 16
--gradient_accumulation_steps 1
--num_train_epochs 1
--save_steps 100
--save_total_limit 5
--learning_rate 1e-6
--fp16
--remove_unused_columns false
--logging_steps 10
--eval_steps 100
--load_best_model_at_end true
--evaluation_strategy "steps"
--output_dir "/data/program/data/dataset6/model" > train.log 2>&1
然后运行infer 通过测试数据做了对比
通过这些指标看效果不好,是参数样本数据少还是样本数据有问题?
楼主,我也遇到这个问题了,准确率也只有百分之30多,请问你是怎么解决的呢