|
28 | 28 | "import pandas as pd" |
29 | 29 | ] |
30 | 30 | }, |
31 | | - |
| 31 | + { |
| 32 | + "cell_type": "code", |
| 33 | + "execution_count": 2, |
| 34 | + "metadata": {}, |
| 35 | + "outputs": [], |
| 36 | + "source": [ |
| 37 | + "# Split the dataframe into test and train data\n", |
| 38 | + "def = split_data(df):\n", |
| 39 | + "X = df.drop('Y', axis=1).values\n", |
| 40 | + "y = df['Y'].values\n", |
| 41 | + "\n", |
| 42 | + "X_train, X_test, y_train, y_test = train_test_split(\n", |
| 43 | + " X, y, test_size=0.2, random_state=0)\n", |
| 44 | + "data = {\"train\": {\"X\": X_train, \"y\": y_train},\n", |
| 45 | + " \"test\": {\"X\": X_test, \"y\": y_test}}\n", |
| 46 | + " return data" |
| 47 | + ] |
| 48 | + }, |
32 | 49 | { |
33 | 50 | "cell_type": "markdown", |
34 | 51 | "metadata": {}, |
|
62 | 79 | "reg_model = Ridge(**args)" |
63 | 80 | ] |
64 | 81 | }, |
65 | | - { |
66 | | - "cell_type": "code", |
67 | | - "execution_count": 2, |
68 | | - "metadata": {}, |
69 | | - "outputs": [], |
70 | | - "source": [ |
71 | | - "# Split the dataframe into test and train data\n", |
72 | | - "def = split_data(df):\n", |
73 | | - "X = df.drop('Y', axis=1).values\n", |
74 | | - "y = df['Y'].values\n", |
75 | | - "\n", |
76 | | - "X_train, X_test, y_train, y_test = train_test_split(\n", |
77 | | - " X, y, test_size=0.2, random_state=0)\n", |
78 | | - "data = {\"train\": {\"X\": X_train, \"y\": y_train},\n", |
79 | | - " \"test\": {\"X\": X_test, \"y\": y_test}}\n", |
80 | | - " return data" |
81 | | - ] |
82 | | - }, |
83 | 82 | { |
84 | 83 | "cell_type": "markdown", |
85 | 84 | "metadata": {}, |
|
0 commit comments