Provide debug data info for floret vectors
@@ -19,6 +19,7 @@ from ..morphology import Morphology
from ..language import Language
from ..util import registry, resolve_dot_names
from ..compat import Literal
+from ..vectors import Mode as VectorsMode
from .. import util
@@ -170,6 +171,14 @@ def debug_data(
show=verbose,
)
if len(nlp.vocab.vectors):
+ if nlp.vocab.vectors.mode == VectorsMode.floret:
+ msg.info(
+ f"floret vectors with {len(nlp.vocab.vectors)} vectors, "
+ f"{nlp.vocab.vectors_length} dimensions, "
+ f"{nlp.vocab.vectors.minn}-{nlp.vocab.vectors.maxn} char "
+ f"n-gram subwords"
+ )
+ else:
msg.info(
f"{len(nlp.vocab.vectors)} vectors ({nlp.vocab.vectors.n_keys} "
f"unique keys, {nlp.vocab.vectors_length} dimensions)"
