| 80 | |
| 81 | |
| 82 | class FMClassifier(BaseFM): |
| 83 | def fit(self, X, y=None): |
| 84 | super(FMClassifier, self).fit(X, y) |
| 85 | self.loss = binary_crossentropy |
| 86 | self.loss_grad = elementwise_grad(binary_crossentropy) |
| 87 | |
| 88 | def predict(self, X=None): |
| 89 | predictions = self._predict(X) |
| 90 | return np.sign(predictions) |
nothing calls this directly
no outgoing calls
no test coverage detected