udellgroup / sketchysgd Goto Github PK

Warning: search size is not used for sketchysgd
optimizer = sketchysgd
dataset = rcv1
dataset location = /home/qys/Research/SketchySGD/data
# of epochs = 20
# of samples in random search = 10
random seeds = [123, 234, 345]
numpy seeds = [246, 468, 690]
model type = logistic
{'precond': 3200}
Traceback (most recent call last):
  File "/home/qys/Research/SketchySGD/general_experiments.py", line 202, in <module>
    main()
  File "/home/qys/Research/SketchySGD/general_experiments.py", line 192, in main
    result = experiment.run(max_epochs, bg)
  File "/home/qys/Research/SketchySGD/experiment.py", line 93, in run
    opt.step(batch)
  File "/home/qys/Research/SketchySGD/opts/sketchysgd.py", line 35, in step
    sing_val = la.svds(lin_op, k = 1, which = 'LM', return_singular_vectors = False)
  File "/home/qys/Research/SketchySGD/.venv/lib/python3.10/site-packages/scipy/sparse/linalg/_eigen/_svds.py", line 539, in svds
    Av = X_matmat(eigvec)
  File "/home/qys/Research/SketchySGD/.venv/lib/python3.10/site-packages/scipy/sparse/linalg/_interface.py", line 377, in rmatmat
    Y = self._rmatmat(X)
  File "/home/qys/Research/SketchySGD/.venv/lib/python3.10/site-packages/scipy/sparse/linalg/_interface.py", line 542, in _rmatmat
    return super()._rmatmat(X)
  File "/home/qys/Research/SketchySGD/.venv/lib/python3.10/site-packages/scipy/sparse/linalg/_interface.py", line 387, in _rmatmat
    return self.H.matmat(X)
  File "/home/qys/Research/SketchySGD/.venv/lib/python3.10/site-packages/scipy/sparse/linalg/_interface.py", line 337, in matmat
    Y = self._matmat(X)
  File "/home/qys/Research/SketchySGD/.venv/lib/python3.10/site-packages/scipy/sparse/linalg/_interface.py", line 527, in _matmat
    return super()._matmat(X)
  File "/home/qys/Research/SketchySGD/.venv/lib/python3.10/site-packages/scipy/sparse/linalg/_interface.py", line 187, in _matmat
    return np.hstack([self.matvec(col.reshape(-1,1)) for col in X.T])
  File "/home/qys/Research/SketchySGD/.venv/lib/python3.10/site-packages/scipy/sparse/linalg/_interface.py", line 187, in <listcomp>
    return np.hstack([self.matvec(col.reshape(-1,1)) for col in X.T])
  File "/home/qys/Research/SketchySGD/.venv/lib/python3.10/site-packages/scipy/sparse/linalg/_interface.py", line 242, in matvec
    y = y.reshape(M,1)
ValueError: cannot reshape array of size 12092416 into shape (47236,1)

$ pipenv run pip list --not-required
Package      Version
------------ -------
fqdn         1.5.1
isoduration  20.11.0
jsonpointer  2.3
jupyterlab   3.6.1
matplotlib   3.7.1
pandas       1.5.3
pip          22.3.1
scikit-learn 1.2.2
setuptools   66.1.1
uri-template 1.2.0
webcolors    1.12
wheel        0.38.4