{ "cells": [ { "cell_type": "markdown", "id": "c49b5055-cb25-4563-966d-430aebe0f434", "metadata": {}, "source": [ "# Salary data" ] }, { "cell_type": "code", "execution_count": 7, "id": "efeae42d-4556-483c-99ae-06e7f81216a2", "metadata": {}, "outputs": [], "source": [ "# Module\n", "import numpy as np\n", "import pandas as pd\n", "import matplotlib.pyplot as plt\n", "\n", "# sklearn\n", "from sklearn.model_selection import train_test_split\n", "from sklearn.linear_model import LinearRegression" ] }, { "cell_type": "code", "execution_count": 8, "id": "7a0abee4-5bb6-4188-9613-94b9755bd8c7", "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", " | YearsExperience | \n", "Salary | \n", "
---|---|---|
0 | \n", "1.1 | \n", "18734.761905 | \n", "
1 | \n", "1.3 | \n", "22002.380952 | \n", "
2 | \n", "1.5 | \n", "17967.142857 | \n", "
3 | \n", "2.0 | \n", "20726.190476 | \n", "
4 | \n", "2.2 | \n", "18995.714286 | \n", "