Open
Description
Hi,
I am wondering if it's possible to add self-defined Embedding logical type into ww which represents vector data? I tried with below code but failed.
import pandas as pd
import numpy as np
import woodwork as ww
from woodwork.logical_types import LogicalType
class Embedding(LogicalType):
primary_dtype = 'object'
standard_tags = {'embedding', 'numeric'}
ww.type_system.add_type(Embedding)
df = pd.DataFrame(
{
"id": [0, 1, 2, 3],
"code": ["012345412359", "122345712358", "012345412359", "012345412359"],
'embedding_0': [np.array([1, 2, 3]), np.array([2, 3, 4]), np.array([3, 4, 5]), np.array([4, 5, 6])],
'embedding_1': [[1, 2, 3], [2, 3, 4], [3, 4, 5], [4, 5, 6]],
}
)
with ww.config.with_options():
df.ww.init()
df.ww
Metadata
Metadata
Assignees
Labels
No labels
Activity