Skip to content

Commit

Permalink
Notebooks explicitly cast int64 (#555)
Browse files Browse the repository at this point in the history
* Explicitly cast to int64 in example notebooks

* Remove 'pdb on' cell from notebook

Was an accidental check-in

Co-authored-by: ottonemo <marian.tietz@ottogroup.com>
  • Loading branch information
BenjaminBossan and ottonemo committed Feb 14, 2020
1 parent 817697f commit 0ec6c3d
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 19 deletions.
2 changes: 1 addition & 1 deletion notebooks/Advanced_Usage.ipynb
Expand Up @@ -103,7 +103,7 @@
"outputs": [],
"source": [
"X, y = make_classification(1000, 20, n_informative=10, random_state=0)\n",
"X = X.astype(np.float32)"
"X, y = X.astype(np.float32), y.astype(np.int64)"
]
},
{
Expand Down
19 changes: 1 addition & 18 deletions notebooks/Basic_Usage.ipynb
Expand Up @@ -119,7 +119,7 @@
"outputs": [],
"source": [
"X, y = make_classification(1000, 20, n_informative=10, random_state=0)\n",
"X = X.astype(np.float32)"
"X, y = X.astype(np.float32), y.astype(np.int64)"
]
},
{
Expand Down Expand Up @@ -236,23 +236,6 @@
"As in `sklearn`, we call `fit` passing the input data `X` and the targets `y`. By default, `NeuralNetClassifier` makes a `StratifiedKFold` split on the data (80/20) to track the validation loss. This is shown, as well as the train loss and the accuracy on the validation set."
]
},
{
"cell_type": "code",
"execution_count": 9,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Automatic pdb calling has been turned ON\n"
]
}
],
"source": [
"pdb on"
]
},
{
"cell_type": "code",
"execution_count": 10,
Expand Down

0 comments on commit 0ec6c3d

Please sign in to comment.