{ "cells": [ { "cell_type": "markdown", "id": "b3d4d608", "metadata": {}, "source": [ "# Optuna一种超参数优化框架\n", "https://github.com/optuna/optuna\n", "\n", "分别采用两组数据集进行比较,分别是加利福尼亚住房数据集(回归)和森林植被类型(多分类)" ] }, { "cell_type": "markdown", "id": "b22b6f90", "metadata": {}, "source": [ "## 结论:效果不错" ] }, { "cell_type": "markdown", "id": "a2d41e62", "metadata": {}, "source": [ "## 波士顿房价预测任务(回归)" ] }, { "cell_type": "code", "execution_count": 1, "id": "f5e0e977", "metadata": {}, "outputs": [], "source": [ "import numpy as np\n", "import pandas as pd\n", "import time\n", "import gc\n", "\n", "from sklearn.model_selection import train_test_split\n", "from sklearn.metrics import mean_squared_error\n", "import lightgbm as lgb # 使用lgb模型" ] }, { "cell_type": "code", "execution_count": 2, "id": "5eda2637", "metadata": {}, "outputs": [], "source": [ "from sklearn.datasets import fetch_california_housing\n", "data = fetch_california_housing()\n", "X, y = data['data'], data['target']" ] }, { "cell_type": "code", "execution_count": 3, "id": "722beb8e", "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", " | MedInc | \n", "HouseAge | \n", "AveRooms | \n", "AveBedrms | \n", "Population | \n", "AveOccup | \n", "Latitude | \n", "Longitude | \n", "
---|---|---|---|---|---|---|---|---|
0 | \n", "8.3252 | \n", "41.0 | \n", "6.984127 | \n", "1.02381 | \n", "322.0 | \n", "2.555556 | \n", "37.88 | \n", "-122.23 | \n", "
1 | \n", "8.3014 | \n", "21.0 | \n", "6.238137 | \n", "0.97188 | \n", "2401.0 | \n", "2.109842 | \n", "37.86 | \n", "-122.22 | \n", "
\n", " | Elevation | \n", "Aspect | \n", "Slope | \n", "Horizontal_Distance_To_Hydrology | \n", "Vertical_Distance_To_Hydrology | \n", "Horizontal_Distance_To_Roadways | \n", "Hillshade_9am | \n", "Hillshade_Noon | \n", "Hillshade_3pm | \n", "Horizontal_Distance_To_Fire_Points | \n", "... | \n", "Soil_Type_30 | \n", "Soil_Type_31 | \n", "Soil_Type_32 | \n", "Soil_Type_33 | \n", "Soil_Type_34 | \n", "Soil_Type_35 | \n", "Soil_Type_36 | \n", "Soil_Type_37 | \n", "Soil_Type_38 | \n", "Soil_Type_39 | \n", "
---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
0 | \n", "2596.0 | \n", "51.0 | \n", "3.0 | \n", "258.0 | \n", "0.0 | \n", "510.0 | \n", "221.0 | \n", "232.0 | \n", "148.0 | \n", "6279.0 | \n", "... | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "
1 | \n", "2590.0 | \n", "56.0 | \n", "2.0 | \n", "212.0 | \n", "-6.0 | \n", "390.0 | \n", "220.0 | \n", "235.0 | \n", "151.0 | \n", "6225.0 | \n", "... | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "
2 rows × 54 columns
\n", "