NameError: name 'Graph2Vec' is not defined

[thucnt]

I installed graph2vec successfully but I cannot use it

import graph2vec.trainer graph2vec = Graph2Vec(vector_dimensions=128) Traceback (most recent call last): File "", line 1, in NameError: name 'Graph2Vec' is not defined

[longchuanren]

from graph2vec.trainer import Graph2Vec