Skip to content
Permalink
Browse files

Notebooks explicitly cast int64 (#555)

* 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 0ec6c3da84053f171fb449b8b0966f61d0c1a383
Showing with 2 additions and 19 deletions.
  1. +1 −1 notebooks/Advanced_Usage.ipynb
  2. +1 −18 notebooks/Basic_Usage.ipynb
@@ -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)"
]
},
{
@@ -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)"
]
},
{
@@ -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,

0 comments on commit 0ec6c3d

Please sign in to comment.
You can’t perform that action at this time.