使用coreml工具转换caffe模型，class_label参数怎么设计？

Question

我正在做一个红绿灯检测的demo，检测后的解析代码如下：
self.predictLabel.text = " none (String(describing: prediction.prob["none"]!)) n green (String(describing: prediction.prob["green"]!)) n red (String(describing: prediction.prob["red"]!))"

但是我不知道对应的classlabel怎么去设计，才能让输出参数可以被我这条语句解析，所以来请教各位大佬！