optimization of train_extension:google similarity