import numpy as np features = np.array([ [120.0, 3.0, 10.0], [90.0, 5.0, 7.0], [150.0, 2.0, 14.0],
import numpy as np embeddings = np.array([ [0.9, 0.1, 0.2], [0.1, 0.8, 0.3], [0.7, 0.2, 0.4],