Added ability to return raw weighted class predictions