MCPcopy Create free account
hub / github.com/LUMIA-Group/MemoryDecoder / interpolate

Method interpolate

knn_utils/saveEmbedMulti.py:235–240  ·  view source on GitHub ↗
(dists, knn_log_probs, lm_log_probs, lmbda)

Source from the content-addressed store, hash-verified

233
234 @staticmethod
235 def interpolate(dists, knn_log_probs, lm_log_probs, lmbda):
236 interpolated = torch.logaddexp(
237 lm_log_probs + np.log(1 - lmbda),
238 knn_log_probs + np.log(lmbda))
239
240 return interpolated
241
242 @staticmethod
243 def get_model_last_layer(model_type):

Callers 1

post_forward_hookMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected