Hi Andreas,
does this support EdgeFeaturesGraphCRF models?
I have tried to call it like:
crf = EdgeFeatureGraphCRF(inference_method="qpbo")
ssvm = OneSlackSSVM(model=crf, n_jobs=1, inference_cache = 100, show_loss_every = 10,switch_to=("ad3",{"branch_and_bound":True}),tol=0.0001)
ssvm.fit(X_train,y_train)
and I get this error:
Traceback (most recent call last):
File "/home/imaging/workspace/virtualenv_ale/libCHunter/source/tools/cnn_colonies_identification/train_edge_graph_crf.py", line 32, in <module>
ssvm.fit(X_train,y_train)
File "/usr/local/lib/python2.7/dist-packages/pystruct/learners/one_slack_ssvm.py", line 429, in fit
joint_feature_gt = self.model.batch_joint_feature(X, Y)
File "/usr/local/lib/python2.7/dist-packages/pystruct/models/base.py", line 40, in batch_joint_feature
joint_feature_ += self.joint_feature(x, y)
File "/usr/local/lib/python2.7/dist-packages/pystruct/models/edge_feature_graph_crf.py", line 190, in joint_feature
class_pair_ind = (y[edges[:, 1]] + self.n_states *
IndexError: too many indices
while if I call with the same dataset:
everything worked like a charm.
Thank you,
Ale