mae和acc

博客介绍了模型编译代码,重点阐述了在模型编译时,设置metrics为acc和mae降低loss值的区别。设置为acc以准确率降低loss值,设置为mae则以平均绝对误差降低loss值。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

model.compile(loss=‘MSE’,
optimizer=adam,
metrics=[‘mae’])  metrics = acc和 mae的区别:
设置为acc,以准确率来降低loss值;设置为mae,以平均绝对误差(mean absolute error)来降低loss值

“best_val_acc = 0.0for epoch in range(training_epochs): model.train() train_loss = 0.0 train_mae = 0.0 with tqdm(total=len(train_loader), desc=f'Epoch {epoch + 1}/{training_epochs}', unit='batch', position=0, leave=True) as pbar: for X_batch, Y_batch in train_loader: X_batch, Y_batch = X_batch.to(device), Y_batch.to(device) optimizer.zero_grad() output = model(X_batch) loss = criterion(output, Y_batch) #训练过程 loss.backward() torch.nn.utils.clip_grad_norm_(model.parameters(), max_norm=1.0) optimizer.step() output_np = output.detach().cpu().numpy() Y_batch_np = Y_batch.detach().cpu().numpy() output_np_flat = output_np.reshape(output_np.shape[0], -1) Y_batch_np_flat = Y_batch_np.reshape(Y_batch_np.shape[0], -1) mae = mean_absolute_error(Y_batch_np_flat, output_np_flat) train_loss += loss.item() train_mae += mae.item() pbar.set_postfix({'loss': f'{loss.item():.4f}', 'mae': f'{mae.item():.4f}'}) pbar.update(1) train_loss /= len(train_loader) train_mae /= len(train_loader) history['train_loss'].append(train_loss) history['train_mae'].append(train_mae) model.eval() val_loss = 0.0 val_mae = 0.0 val_acc = 0.0 with torch.no_grad(): for X_batch, Y_batch in valid_loader: X_batch, Y_batch = X_batch.to(device), Y_batch.to(device) output = model(X_batch) loss = criterion(output, Y_batch) output_np = output.detach().cpu().numpy() Y_batch_np = Y_batch.detach().cpu().numpy() output_np_flat = output_np.reshape(output_np.shape[0], -1) Y_batch_np_flat = Y_batch_np.reshape(Y_batch_np.shape[0], -1) mae = mean_absolute_error(Y_batch_np_flat, output_np_flat) val_loss += loss.item() val_mae += mae.item() acc = precision_score(Y_batch_np_flat, output_np_flat) val_acc += acc.item() val_loss /= len(valid_loader) val_mae /= len(valid_loader) val_acc /= len(valid_loader) history['val_loss'].append(val_loss) history['val_mae'].append(val_mae) print( f'Epoch {epoch + 1}/{training_epochs}, Train Loss: {train_loss:.4f}, Validation Loss: {val_loss:.4f}, Validation MAE: {val_mae:.4f}') if val_acc > best_val_acc: best_val_acc = val_acc torch.save(model.state_dict(), file_weights) print(f'Best model saved at epoch {epoch + 1} with validation loss {val_loss:.4f}')”你看我保存参数的定义对吧,如果按照验证机精度来保存的话
最新发布
03-09
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值