alteryx / alteryx/woodwork

Define `Embedding` logical type for vector data

未关闭
#1,871 0 条评论 0 个 reaction 已指派 0 人 在 GitHub 查看
主要语言
Python
星标
155
派生
24
PR 合并指标
30 天内没有已合并 PR

描述

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
```

贡献指南

打开贡献指南

评估

这个 Issue 还没有评估数据。

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。