423 lines
8.8 KiB
JSON
423 lines
8.8 KiB
JSON
|
{
|
||
|
|
"best_config": {
|
||
|
|
"gru_weight": 1.0,
|
||
|
|
"lstm_weight": 0.0,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.0,
|
||
|
|
"scale": 0.5,
|
||
|
|
"shift": 0.5,
|
||
|
|
"smooth": 0.0
|
||
|
|
},
|
||
|
|
"per": 10.122729029764205
|
||
|
|
},
|
||
|
|
"all_results": [
|
||
|
|
{
|
||
|
|
"gru_weight": 0.8,
|
||
|
|
"lstm_weight": 0.19999999999999996,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.0,
|
||
|
|
"scale": 0.0,
|
||
|
|
"shift": 0.0,
|
||
|
|
"smooth": 0.0
|
||
|
|
},
|
||
|
|
"per": 10.45129493621956
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 0.8,
|
||
|
|
"lstm_weight": 0.19999999999999996,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.0,
|
||
|
|
"scale": 0.0,
|
||
|
|
"shift": 0.0,
|
||
|
|
"smooth": 0.5
|
||
|
|
},
|
||
|
|
"per": 10.794356397371473
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 0.8,
|
||
|
|
"lstm_weight": 0.19999999999999996,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.0,
|
||
|
|
"scale": 0.0,
|
||
|
|
"shift": 0.5,
|
||
|
|
"smooth": 0.0
|
||
|
|
},
|
||
|
|
"per": 11.620603015075377
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 0.8,
|
||
|
|
"lstm_weight": 0.19999999999999996,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.0,
|
||
|
|
"scale": 0.0,
|
||
|
|
"shift": 0.5,
|
||
|
|
"smooth": 0.5
|
||
|
|
},
|
||
|
|
"per": 11.850115964437572
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 0.8,
|
||
|
|
"lstm_weight": 0.19999999999999996,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.0,
|
||
|
|
"scale": 0.5,
|
||
|
|
"shift": 0.0,
|
||
|
|
"smooth": 0.0
|
||
|
|
},
|
||
|
|
"per": 10.470622342481638
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 0.8,
|
||
|
|
"lstm_weight": 0.19999999999999996,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.0,
|
||
|
|
"scale": 0.5,
|
||
|
|
"shift": 0.0,
|
||
|
|
"smooth": 0.5
|
||
|
|
},
|
||
|
|
"per": 10.760533436412834
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 0.8,
|
||
|
|
"lstm_weight": 0.19999999999999996,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.0,
|
||
|
|
"scale": 0.5,
|
||
|
|
"shift": 0.5,
|
||
|
|
"smooth": 0.0
|
||
|
|
},
|
||
|
|
"per": 11.42974487823734
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 0.8,
|
||
|
|
"lstm_weight": 0.19999999999999996,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.0,
|
||
|
|
"scale": 0.5,
|
||
|
|
"shift": 0.5,
|
||
|
|
"smooth": 0.5
|
||
|
|
},
|
||
|
|
"per": 11.569868573637418
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 0.8,
|
||
|
|
"lstm_weight": 0.19999999999999996,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.5,
|
||
|
|
"scale": 0.0,
|
||
|
|
"shift": 0.0,
|
||
|
|
"smooth": 0.0
|
||
|
|
},
|
||
|
|
"per": 10.463374565133359
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 0.8,
|
||
|
|
"lstm_weight": 0.19999999999999996,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.5,
|
||
|
|
"scale": 0.0,
|
||
|
|
"shift": 0.0,
|
||
|
|
"smooth": 0.5
|
||
|
|
},
|
||
|
|
"per": 10.772613065326633
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 0.8,
|
||
|
|
"lstm_weight": 0.19999999999999996,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.5,
|
||
|
|
"scale": 0.0,
|
||
|
|
"shift": 0.5,
|
||
|
|
"smooth": 0.0
|
||
|
|
},
|
||
|
|
"per": 11.42249710088906
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 0.8,
|
||
|
|
"lstm_weight": 0.19999999999999996,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.5,
|
||
|
|
"scale": 0.5,
|
||
|
|
"shift": 0.0,
|
||
|
|
"smooth": 0.0
|
||
|
|
},
|
||
|
|
"per": 10.485117897178199
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 0.8,
|
||
|
|
"lstm_weight": 0.19999999999999996,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.5,
|
||
|
|
"scale": 0.0,
|
||
|
|
"shift": 0.5,
|
||
|
|
"smooth": 0.5
|
||
|
|
},
|
||
|
|
"per": 11.565036722071898
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 0.8,
|
||
|
|
"lstm_weight": 0.19999999999999996,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.5,
|
||
|
|
"scale": 0.5,
|
||
|
|
"shift": 0.0,
|
||
|
|
"smooth": 0.5
|
||
|
|
},
|
||
|
|
"per": 10.753285659064554
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 1.0,
|
||
|
|
"lstm_weight": 0.0,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.0,
|
||
|
|
"scale": 0.0,
|
||
|
|
"shift": 0.0,
|
||
|
|
"smooth": 0.0
|
||
|
|
},
|
||
|
|
"per": 10.183127174333205
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 0.8,
|
||
|
|
"lstm_weight": 0.19999999999999996,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.5,
|
||
|
|
"scale": 0.5,
|
||
|
|
"shift": 0.5,
|
||
|
|
"smooth": 0.0
|
||
|
|
},
|
||
|
|
"per": 11.379010436799382
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 1.0,
|
||
|
|
"lstm_weight": 0.0,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.0,
|
||
|
|
"scale": 0.0,
|
||
|
|
"shift": 0.0,
|
||
|
|
"smooth": 0.5
|
||
|
|
},
|
||
|
|
"per": 10.183127174333205
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 1.0,
|
||
|
|
"lstm_weight": 0.0,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.0,
|
||
|
|
"scale": 0.0,
|
||
|
|
"shift": 0.5,
|
||
|
|
"smooth": 0.0
|
||
|
|
},
|
||
|
|
"per": 10.134808658678006
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 0.8,
|
||
|
|
"lstm_weight": 0.19999999999999996,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.5,
|
||
|
|
"scale": 0.5,
|
||
|
|
"shift": 0.5,
|
||
|
|
"smooth": 0.5
|
||
|
|
},
|
||
|
|
"per": 11.48289524545806
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 1.0,
|
||
|
|
"lstm_weight": 0.0,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.0,
|
||
|
|
"scale": 0.5,
|
||
|
|
"shift": 0.0,
|
||
|
|
"smooth": 0.0
|
||
|
|
},
|
||
|
|
"per": 10.173463471202165
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 1.0,
|
||
|
|
"lstm_weight": 0.0,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.0,
|
||
|
|
"scale": 0.0,
|
||
|
|
"shift": 0.5,
|
||
|
|
"smooth": 0.5
|
||
|
|
},
|
||
|
|
"per": 10.41747197526092
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 1.0,
|
||
|
|
"lstm_weight": 0.0,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.0,
|
||
|
|
"scale": 0.5,
|
||
|
|
"shift": 0.0,
|
||
|
|
"smooth": 0.5
|
||
|
|
},
|
||
|
|
"per": 10.168631619636646
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 1.0,
|
||
|
|
"lstm_weight": 0.0,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.0,
|
||
|
|
"scale": 0.5,
|
||
|
|
"shift": 0.5,
|
||
|
|
"smooth": 0.0
|
||
|
|
},
|
||
|
|
"per": 10.122729029764205
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 1.0,
|
||
|
|
"lstm_weight": 0.0,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.5,
|
||
|
|
"scale": 0.0,
|
||
|
|
"shift": 0.0,
|
||
|
|
"smooth": 0.0
|
||
|
|
},
|
||
|
|
"per": 10.185543100115964
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 1.0,
|
||
|
|
"lstm_weight": 0.0,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.0,
|
||
|
|
"scale": 0.5,
|
||
|
|
"shift": 0.5,
|
||
|
|
"smooth": 0.5
|
||
|
|
},
|
||
|
|
"per": 10.190374951681484
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 1.0,
|
||
|
|
"lstm_weight": 0.0,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.5,
|
||
|
|
"scale": 0.0,
|
||
|
|
"shift": 0.0,
|
||
|
|
"smooth": 0.5
|
||
|
|
},
|
||
|
|
"per": 10.200038654812523
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 1.0,
|
||
|
|
"lstm_weight": 0.0,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.5,
|
||
|
|
"scale": 0.0,
|
||
|
|
"shift": 0.5,
|
||
|
|
"smooth": 0.0
|
||
|
|
},
|
||
|
|
"per": 10.127560881329726
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 1.0,
|
||
|
|
"lstm_weight": 0.0,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.5,
|
||
|
|
"scale": 0.0,
|
||
|
|
"shift": 0.5,
|
||
|
|
"smooth": 0.5
|
||
|
|
},
|
||
|
|
"per": 10.192790877464244
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 1.0,
|
||
|
|
"lstm_weight": 0.0,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.5,
|
||
|
|
"scale": 0.5,
|
||
|
|
"shift": 0.0,
|
||
|
|
"smooth": 0.0
|
||
|
|
},
|
||
|
|
"per": 10.168631619636646
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 1.0,
|
||
|
|
"lstm_weight": 0.0,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.5,
|
||
|
|
"scale": 0.5,
|
||
|
|
"shift": 0.0,
|
||
|
|
"smooth": 0.5
|
||
|
|
},
|
||
|
|
"per": 10.183127174333205
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 1.0,
|
||
|
|
"lstm_weight": 0.0,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.5,
|
||
|
|
"scale": 0.5,
|
||
|
|
"shift": 0.5,
|
||
|
|
"smooth": 0.0
|
||
|
|
},
|
||
|
|
"per": 10.137224584460766
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"gru_weight": 1.0,
|
||
|
|
"lstm_weight": 0.0,
|
||
|
|
"tta_weights": {
|
||
|
|
"original": 1.0,
|
||
|
|
"noise": 0.5,
|
||
|
|
"scale": 0.5,
|
||
|
|
"shift": 0.5,
|
||
|
|
"smooth": 0.5
|
||
|
|
},
|
||
|
|
"per": 10.161383842288364
|
||
|
|
}
|
||
|
|
],
|
||
|
|
"args": {
|
||
|
|
"gru_model_path": "/root/autodl-tmp/nejm-brain-to-text/data/t15_pretrained_rnn_baseline",
|
||
|
|
"lstm_model_path": "/root/autodl-tmp/nejm-brain-to-text/model_training_lstm/trained_models/baseline_rnn",
|
||
|
|
"data_dir": "../data/hdf5_data_final",
|
||
|
|
"csv_path": "../data/t15_copyTaskData_description.csv",
|
||
|
|
"eval_type": "val",
|
||
|
|
"gpu_number": 0,
|
||
|
|
"gru_weights": "0.8,1.0",
|
||
|
|
"tta_noise_weights": "0.0,0.5",
|
||
|
|
"tta_scale_weights": "0.0,0.5",
|
||
|
|
"tta_shift_weights": "0.0,0.5",
|
||
|
|
"tta_smooth_weights": "0.0,0.5",
|
||
|
|
"tta_noise_std": 0.01,
|
||
|
|
"tta_smooth_range": 0.5,
|
||
|
|
"tta_scale_range": 0.05,
|
||
|
|
"tta_cut_max": 3,
|
||
|
|
"cache_file": "tta_cache.pkl",
|
||
|
|
"force_recache": false,
|
||
|
|
"output_file": "test_results.json",
|
||
|
|
"max_workers": 4,
|
||
|
|
"batch_size": 100
|
||
|
|
},
|
||
|
|
"timestamp": "2025-09-17 22:25:20"
|
||
|
|
}
|