critical bug "this->train_data[0].size" fixed