Skip to content

Commit

Permalink
tox fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
pranmod01 committed Oct 23, 2024
1 parent 7439e49 commit cfc0045
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/nemos/glm.py
Original file line number Diff line number Diff line change
Expand Up @@ -1474,7 +1474,7 @@ def fit(
>>> num_samples, num_features, num_neurons = 100, 3, 2
>>> X = np.random.normal(size=(num_samples, num_features))
>>> # Weights is defined by how each feature influences the output, shape (num_features, num_neurons)
>>> weights = np.array([[ 0.5, 0. ], [-0.5, -0.5], [ 0. , 1. ]])
>>> weights = np.array([[ 0.5, 0. ], [-0.5, -0.5], [ 0. , 1. ]])
>>> # Output y simulates a Poisson distribution based on a linear model between features X and wegihts
>>> y = np.random.poisson(np.exp(X.dot(weights)))
Expand Down

0 comments on commit cfc0045

Please sign in to comment.