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