Can AI choose the best method of water treatment?