Question

我知道numba现在支持字符串，但是找不到有关如何在jitclass中使用字符串的任何文档，并且无法弄清楚。

如何使用jitclass创建字符串属性？

（此hack是对字符串的预先支持，相当混乱：How can I pass string type in class in numba jitclass python?）

我尝试了unicode_type，char，char [：]，uint8，str-基本上我能想到的一切。

COND_SPEC = [
    ('feature',nb.unicode_type),
    ('val', nb.unicode_type)
]

@jitclass(COND_SPEC)
class Cond:
    """ Class implementing conditional. """

    def __init__(self, feature, val):
        self.feature = feature
        self.val = val

该类可以编译，但是声明该类的实例会产生错误：

c = Cond('education','HS-grad')

numba.errors.LoweringError: Failed in nopython mode pipeline (step: nopython mode backend)
Cannot cast unicode_type to int8: %".37" = load {i8*, i64, i32, i64, i8*, i8*}, {i8*, i64, i32, i64, i8*, i8*}* %"feature"

File "<ipython-input-19-aaeb1c1955cb>", line 12:
    def __init__(self, feature, val):
        self.feature = feature
        ^

[1] During: lowering "(self).feature = feature" at <ipython-input-19-aaeb1c1955cb> (12)
[2] During: resolving callee type: jitclass.Cond#7f9c36758a18<feature:int8,val:int8>
[3] During: typing of call at <string> (3)

--%<----------------------------------------------------------------------------


File "<string>", line 3:
<source missing, REPL/exec in use?>

Answer 1

我相信这是numba.types.string：

import numba as nb
from numba import jitclass

COND_SPEC = [
    ('feature',nb.types.string),
    ('val', nb.types.string)
]

@jitclass(COND_SPEC)
class Cond:
    """ Class implementing conditional. """

    def __init__(self, feature, val):
        self.feature = feature
        self.val = val

c = Cond('Hello', 'world')
print(c.feature, c.val)

>>>Hello world

有没有办法在numba jitclass中包含字符串属性？

1 个答案: