Table 1 Performance of different models for graphene property prediction.

From: Efficient water desalination with graphene nanopores obtained using artificial intelligence

Model

Flux MSE

Flux R2

Ion rejection MSE

Ion rejection R2

XGBoost

0.011

0.988

0.008

0.992

VGG16

0.0448

0.957

0.0156

0.985

ResNet18

0.0024

0.998

0.0039

0.996

ResNet50

0.0022

0.998

0.0038

0.996

  1. Values based on standardized water flux/ion rejection.