我正在尝试使用此处的代码:https ://github.com/pystruct/pystruct/blob/master/examples/multi_label.py
我有 X_train 与 shape(2591, 256)
和 y_train 与 shape (2591, 175)
。当我运行这个:
tree = chow_liu_tree(y_train)
tree_model = MultiLabelClf(edges=tree, inference_method="max-product")
tree_ssvm = OneSlackSSVM(tree_model, inference_cache=50, C=.1, tol=0.01)
print("fitting tree model...")
tree_ssvm.fit(X_train, y_train)
我懂了:
Traceback (most recent call last):
File "classifiers.py", line 173, in <module>
tree_ssvm.fit(X_train, y_train)
File "/usr/local/lib/python2.7/dist-packages/pystruct/learners/one_slack_ssvm.py", line 448, in fit
X, Y, joint_feature_gt, constraints)
File "/usr/local/lib/python2.7/dist-packages/pystruct/learners/one_slack_ssvm.py", line 348, in _find_new_constraint
X, Y, self.w, relaxed=True)
File "/usr/local/lib/python2.7/dist-packages/pystruct/models/base.py", line 95, in batch_loss_augmented_inference
for x, y in zip(X, Y)]
File "/usr/local/lib/python2.7/dist-packages/pystruct/models/crf.py", line 106, in loss_augmented_inference
loss_augment_unaries(unary_potentials, np.asarray(y), self.class_weight)
File "utils.pyx", line 21, in utils.__pyx_fused_cpdef (src/utils.c:4341)
TypeError: No matching signature found
当我直接从链接运行代码时,它可以工作(使用他们的数据集)。有谁知道应该是什么问题?