Corrects node size

[yutsai84]

trafficstars

The existing implementation assumed the max of node_indicies is the size of the nodes.

The assumption is not alway true. For example, if the largest node index of the nodes is not in node_indicies, this assumption is wrong.

This PR addresses this by directly checking the size of nodes. That is the size of the nodes is the row number of node_representations.