Skip to content
Snippets Groups Projects
Commit bd922416 authored by Jörg Martin's avatar Jörg Martin
Browse files

Metrics for coverage and bias included

Several metrics were included in evaluate_tabular.py. To this end
the file coverage_metrices.py was added and the processing of a larger
number of metrics in evaluate_tabular.py was simplified.
parents ddee5e7e f58054ef
No related branches found
No related tags found
No related merge requests found
......@@ -69,6 +69,8 @@ class UpdatedTrainEpoch(train_and_store.TrainEpoch):
rmse = self.rmse(net).item()
rmse_chain.append(rmse)
writer.add_scalar('RMSE', rmse, self.total_count)
writer.add_scalar('std_y', self.last_std_y, self.total_count)
writer.add_scalar('RMSE:std_y', rmse/self.last_std_y, self.total_count)
writer.add_scalar('train loss', self.last_train_loss, self.total_count)
writer.add_scalar('test loss', self.last_test_loss, self.total_count)
print(f'RMSE {rmse:.3f}')
......
......@@ -69,6 +69,8 @@ class UpdatedTrainEpoch(train_and_store.TrainEpoch):
rmse = self.rmse(net).item()
rmse_chain.append(rmse)
writer.add_scalar('RMSE', rmse, self.total_count)
writer.add_scalar('std_y', self.last_std_y, self.total_count)
writer.add_scalar('RMSE:std_y', rmse/self.last_std_y, self.total_count)
writer.add_scalar('train loss', self.last_train_loss, self.total_count)
writer.add_scalar('test loss', self.last_test_loss, self.total_count)
print(f'RMSE {rmse:.3f}')
......
......@@ -69,6 +69,8 @@ class UpdatedTrainEpoch(train_and_store.TrainEpoch):
rmse = self.rmse(net).item()
rmse_chain.append(rmse)
writer.add_scalar('RMSE', rmse, self.total_count)
writer.add_scalar('std_y', self.last_std_y, self.total_count)
writer.add_scalar('RMSE:std_y', rmse/self.last_std_y, self.total_count)
writer.add_scalar('train loss', self.last_train_loss, self.total_count)
writer.add_scalar('test loss', self.last_test_loss, self.total_count)
print(f'RMSE {rmse:.3f}')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment