|
6610 | 6610 | "source": [ |
6611 | 6611 | "from spotpython.build.kriging import Kriging\n", |
6612 | 6612 | "import numpy as np\n", |
6613 | | - "import matplotlib.pyplot as plt\n", |
6614 | 6613 | "from numpy import linspace, arange\n", |
6615 | 6614 | "rng = np.random.RandomState(1)\n", |
6616 | 6615 | "X = linspace(start=0, stop=10, num=1_0).reshape(-1, 1)\n", |
|
6640 | 6639 | "source": [ |
6641 | 6640 | "from spotpython.build.kriging import Kriging\n", |
6642 | 6641 | "import numpy as np\n", |
6643 | | - "import matplotlib.pyplot as plt\n", |
6644 | | - "from numpy import linspace, arange, empty\n", |
| 6642 | + " from numpy import linspace, arange, empty\n", |
6645 | 6643 | "rng = np.random.RandomState(1)\n", |
6646 | 6644 | "X = linspace(start=0, stop=10, num=10).reshape(-1, 1)\n", |
6647 | 6645 | "y = np.squeeze(X * np.sin(X))\n", |
|
6654 | 6652 | "s = empty(n, dtype=float)\n", |
6655 | 6653 | "ei = empty(n, dtype=float)\n", |
6656 | 6654 | "for i in range(n):\n", |
6657 | | - " x = X[i, :]\n", |
6658 | | - " y[i], s[i], ei[i] = S.predict_coded(x)\n", |
| 6655 | + " y_coded, s_coded, ei_coded = S.predict_coded(X[i, :])\n", |
| 6656 | + " y[i] = y_coded if np.isscalar(y_coded) else y_coded.item()\n", |
| 6657 | + " s[i] = s_coded if np.isscalar(s_coded) else s_coded.item()\n", |
| 6658 | + " ei[i] = ei_coded if np.isscalar(ei_coded) else ei_coded.item()\n", |
6659 | 6659 | "print(f\"y: {y}\")\n", |
6660 | 6660 | "print(f\"s: {s}\")\n", |
6661 | 6661 | "print(f\"ei: {-1.0*ei}\")" |
|
6670 | 6670 | }, |
6671 | 6671 | { |
6672 | 6672 | "cell_type": "code", |
6673 | | - "execution_count": null, |
| 6673 | + "execution_count": 1, |
6674 | 6674 | "metadata": {}, |
6675 | | - "outputs": [], |
| 6675 | + "outputs": [ |
| 6676 | + { |
| 6677 | + "name": "stdout", |
| 6678 | + "output_type": "stream", |
| 6679 | + "text": [ |
| 6680 | + "S.psi: [[1.83156389e-02]\n", |
| 6681 | + " [4.13993772e-08]\n", |
| 6682 | + " [4.24835426e-18]]\n", |
| 6683 | + "Control value res: [[1.83156389e-02]\n", |
| 6684 | + " [4.13993772e-08]\n", |
| 6685 | + " [4.24835426e-18]]\n" |
| 6686 | + ] |
| 6687 | + } |
| 6688 | + ], |
6676 | 6689 | "source": [ |
6677 | 6690 | "import numpy as np\n", |
6678 | 6691 | "from spotpython.build.kriging import Kriging\n", |
|
0 commit comments