From 4d16d9610ceb32686684ba01ac9876bce6a57cac Mon Sep 17 00:00:00 2001 From: Stefan Hahmann Date: Fri, 15 Nov 2024 17:21:58 +0100 Subject: [PATCH] Do not show any dimensionality reduction outputs as input for dimensionality reduction --- .../feature/dimensionalityreduction/util/InputDimension.java | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/main/java/org/mastodon/mamut/feature/dimensionalityreduction/util/InputDimension.java b/src/main/java/org/mastodon/mamut/feature/dimensionalityreduction/util/InputDimension.java index f8b54670c..0dd7441db 100644 --- a/src/main/java/org/mastodon/mamut/feature/dimensionalityreduction/util/InputDimension.java +++ b/src/main/java/org/mastodon/mamut/feature/dimensionalityreduction/util/InputDimension.java @@ -38,7 +38,7 @@ import org.mastodon.graph.Vertex; import org.mastodon.mamut.feature.LinkTargetIdFeature; import org.mastodon.mamut.feature.SpotTrackIDFeature; -import org.mastodon.mamut.feature.dimensionalityreduction.umap.feature.AbstractUmapFeature; +import org.mastodon.mamut.feature.dimensionalityreduction.AbstractOutputFeature; import org.mastodon.mamut.feature.spot.SpotBranchIDFeature; import org.mastodon.util.FeatureUtils; @@ -162,7 +162,7 @@ public static < V extends Vertex< E >, E extends Edge< V > > List< InputDimensio excludedVertexFeatures.add( Cast.unchecked( SpotBranchIDFeature.class ) ); for ( Feature< V > feature : vertexFeatures ) { - if ( excludedVertexFeatures.contains( feature.getClass() ) || feature instanceof AbstractUmapFeature ) + if ( excludedVertexFeatures.contains( feature.getClass() ) || feature instanceof AbstractOutputFeature ) continue; for ( FeatureProjection< V > projection : feature.projections() ) inputDimensions.add( InputDimension.fromVertexFeature( feature, projection ) );