8d0d951df39bfaafdcb7951fb0d79dd21caf6ae0,fastFM/sgd.py,FMClassification,fit,#FMClassification#Any#Any#,146

Before Change

        
        y_train = _validate_class_labels(y_train)
        assert_all_finite(X_train)
        assert_all_finite(y_train)

        self.w0_, self.w_, self.V_ = ffm.ffm_sgd_fit(self, X_train, y_train)
        return self

After Change

        check_consistent_length(X, y)
        y = y.astype(np.float64)
        X = X.T
        X = check_array(X, accept_sparse="csc", dtype=np.float64)

        self.w0_, self.w_, self.V_ = ffm.ffm_sgd_fit(self, X, y)
        return self

Instances