Update similarity.py

[sparkingarthur]

update predict method to support batch prediction.

添加对于批量预测的支持。使用方式与原来略有改变，使用及测试见main函数。