Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- {
- "cells": [
- {
- "cell_type": "code",
- "execution_count": 4,
- "metadata": {},
- "outputs": [
- {
- "data": {
- "text/plain": [
- "'0.20.3'"
- ]
- },
- "execution_count": 4,
- "metadata": {},
- "output_type": "execute_result"
- }
- ],
- "source": [
- "import pandas as pd\n",
- "import numpy as np\n",
- "import sklearn\n",
- "from sklearn import linear_model, ensemble\n",
- "from sklearn.model_selection import train_test_split\n",
- "from sklearn.utils import shuffle\n",
- "import copy\n",
- "sklearn.__version__"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": 5,
- "metadata": {},
- "outputs": [
- {
- "data": {
- "text/html": [
- "<div>\n",
- "<style scoped>\n",
- " .dataframe tbody tr th:only-of-type {\n",
- " vertical-align: middle;\n",
- " }\n",
- "\n",
- " .dataframe tbody tr th {\n",
- " vertical-align: top;\n",
- " }\n",
- "\n",
- " .dataframe thead th {\n",
- " text-align: right;\n",
- " }\n",
- "</style>\n",
- "<table border=\"1\" class=\"dataframe\">\n",
- " <thead>\n",
- " <tr style=\"text-align: right;\">\n",
- " <th></th>\n",
- " <th>age</th>\n",
- " <th>workclass</th>\n",
- " <th>fnlwgt</th>\n",
- " <th>education</th>\n",
- " <th>education-num</th>\n",
- " <th>marital-status</th>\n",
- " <th>occupation</th>\n",
- " <th>relationship</th>\n",
- " <th>race</th>\n",
- " <th>sex</th>\n",
- " <th>capital-gain</th>\n",
- " <th>capital-loss</th>\n",
- " <th>hours-per-week</th>\n",
- " <th>native-country</th>\n",
- " <th>yearly-income</th>\n",
- " </tr>\n",
- " </thead>\n",
- " <tbody>\n",
- " <tr>\n",
- " <th>0</th>\n",
- " <td>39</td>\n",
- " <td>State-gov</td>\n",
- " <td>77516</td>\n",
- " <td>Bachelors</td>\n",
- " <td>13</td>\n",
- " <td>Never-married</td>\n",
- " <td>Adm-clerical</td>\n",
- " <td>Not-in-family</td>\n",
- " <td>White</td>\n",
- " <td>Male</td>\n",
- " <td>2174</td>\n",
- " <td>0</td>\n",
- " <td>40</td>\n",
- " <td>United-States</td>\n",
- " <td><=50K</td>\n",
- " </tr>\n",
- " <tr>\n",
- " <th>1</th>\n",
- " <td>50</td>\n",
- " <td>Self-emp-not-inc</td>\n",
- " <td>83311</td>\n",
- " <td>Bachelors</td>\n",
- " <td>13</td>\n",
- " <td>Married-civ-spouse</td>\n",
- " <td>Exec-managerial</td>\n",
- " <td>Husband</td>\n",
- " <td>White</td>\n",
- " <td>Male</td>\n",
- " <td>0</td>\n",
- " <td>0</td>\n",
- " <td>13</td>\n",
- " <td>United-States</td>\n",
- " <td><=50K</td>\n",
- " </tr>\n",
- " <tr>\n",
- " <th>2</th>\n",
- " <td>38</td>\n",
- " <td>Private</td>\n",
- " <td>215646</td>\n",
- " <td>HS-grad</td>\n",
- " <td>9</td>\n",
- " <td>Divorced</td>\n",
- " <td>Handlers-cleaners</td>\n",
- " <td>Not-in-family</td>\n",
- " <td>White</td>\n",
- " <td>Male</td>\n",
- " <td>0</td>\n",
- " <td>0</td>\n",
- " <td>40</td>\n",
- " <td>United-States</td>\n",
- " <td><=50K</td>\n",
- " </tr>\n",
- " <tr>\n",
- " <th>3</th>\n",
- " <td>53</td>\n",
- " <td>Private</td>\n",
- " <td>234721</td>\n",
- " <td>11th</td>\n",
- " <td>7</td>\n",
- " <td>Married-civ-spouse</td>\n",
- " <td>Handlers-cleaners</td>\n",
- " <td>Husband</td>\n",
- " <td>Black</td>\n",
- " <td>Male</td>\n",
- " <td>0</td>\n",
- " <td>0</td>\n",
- " <td>40</td>\n",
- " <td>United-States</td>\n",
- " <td><=50K</td>\n",
- " </tr>\n",
- " <tr>\n",
- " <th>4</th>\n",
- " <td>28</td>\n",
- " <td>Private</td>\n",
- " <td>338409</td>\n",
- " <td>Bachelors</td>\n",
- " <td>13</td>\n",
- " <td>Married-civ-spouse</td>\n",
- " <td>Prof-specialty</td>\n",
- " <td>Wife</td>\n",
- " <td>Black</td>\n",
- " <td>Female</td>\n",
- " <td>0</td>\n",
- " <td>0</td>\n",
- " <td>40</td>\n",
- " <td>Cuba</td>\n",
- " <td><=50K</td>\n",
- " </tr>\n",
- " </tbody>\n",
- "</table>\n",
- "</div>"
- ],
- "text/plain": [
- " age workclass fnlwgt education education-num \\\n",
- "0 39 State-gov 77516 Bachelors 13 \n",
- "1 50 Self-emp-not-inc 83311 Bachelors 13 \n",
- "2 38 Private 215646 HS-grad 9 \n",
- "3 53 Private 234721 11th 7 \n",
- "4 28 Private 338409 Bachelors 13 \n",
- "\n",
- " marital-status occupation relationship race sex \\\n",
- "0 Never-married Adm-clerical Not-in-family White Male \n",
- "1 Married-civ-spouse Exec-managerial Husband White Male \n",
- "2 Divorced Handlers-cleaners Not-in-family White Male \n",
- "3 Married-civ-spouse Handlers-cleaners Husband Black Male \n",
- "4 Married-civ-spouse Prof-specialty Wife Black Female \n",
- "\n",
- " capital-gain capital-loss hours-per-week native-country yearly-income \n",
- "0 2174 0 40 United-States <=50K \n",
- "1 0 0 13 United-States <=50K \n",
- "2 0 0 40 United-States <=50K \n",
- "3 0 0 40 United-States <=50K \n",
- "4 0 0 40 Cuba <=50K "
- ]
- },
- "execution_count": 5,
- "metadata": {},
- "output_type": "execute_result"
- }
- ],
- "source": [
- "df = pd.read_csv(\"income.data\", header = 0)\n",
- "df.columns = [col.strip() for col in df.columns]\n",
- "df.head()"
- ]
- }
- ],
- "metadata": {
- "kernelspec": {
- "display_name": "Python 3",
- "language": "python",
- "name": "python3"
- },
- "language_info": {
- "codemirror_mode": {
- "name": "ipython",
- "version": 3
- },
- "file_extension": ".py",
- "mimetype": "text/x-python",
- "name": "python",
- "nbconvert_exporter": "python",
- "pygments_lexer": "ipython3",
- "version": "3.6.5"
- }
- },
- "nbformat": 4,
- "nbformat_minor": 2
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement