how to export knowledgegraph from neo4j with just the required number of rows?

I used Neo4j to build a knowledge graph on a dataset containing 200 rows. Now the output from knowledge graph after building all the relations and entity is 4000 rows which I need to use as input for my transE model to get the embedded vectors.The problem is I need to link the embedded vectors to the class label containing only 200 values. Is there a way here to get the output from neo4j with just 200 rows??