{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"pygments_lexer":"ipython3","nbconvert_exporter":"python","version":"3.6.4","file_extension":".py","codemirror_mode":{"name":"ipython","version":3},"name":"python","mimetype":"text/x-python"}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"markdown","source":"# Install MLJAR and Libraries","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19"}},{"cell_type":"code","source":"!pip install -q -U git+https://github.com/mljar/mljar-supervised.git@master","metadata":{},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"import numpy as np\nimport pandas as pd\nfrom supervised.automl import AutoML ","metadata":{},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# read files","metadata":{}},{"cell_type":"code","source":"train = pd.read_csv(\"../input/tabular-playground-series-aug-2022/train.csv\")\ntest = pd.read_csv(\"../input/tabular-playground-series-aug-2022/test.csv\")","metadata":{},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"x_cols = train.columns[1:-1].tolist()\ny_col = train.columns[-1]","metadata":{},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# AutoML it!","metadata":{}},{"cell_type":"code","source":"automl = AutoML(\n    mode=\"Compete\", \n    total_time_limit=6*60*60, # 6 hours \n    eval_metric='auc'\n)\nautoml.fit(train[x_cols], train[y_col])","metadata":{},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Submission","metadata":{}},{"cell_type":"code","source":"preds = automl.predict_proba(test)\nsub = pd.DataFrame({\"id\": test.id, \"failure\": preds[:,1]})\nsub.to_csv(\"./1_sub.csv\", index=False)","metadata":{},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Report","metadata":{}},{"cell_type":"code","source":"automl.report()","metadata":{},"execution_count":null,"outputs":[]}]}