aertslab / CREsted

Other
26 stars 1 forks source link

get_embeddings based on layer name #17

Closed LukasMahieu closed 1 month ago

LukasMahieu commented 1 month ago

Adds a function to the Crested object to return the embeddings based on the model's layer name. Useful for visualizing representations.

Example

evaluator = crested.tl.Crested(....)
evaluator.load_model(...)
embeddings = evaluator.get_embeddings(layer_name="global_average_pooling1d_3")