notebook??

examples/ensemble/mrbbagging_pipeline.ipynb

@@ -7,7 +7,21 @@
  "collapsed": true
  },
  "outputs": [],
- "source": []
+ "source": [
+ "from multi_imbalance.ensemble.mrbbagging import MRBBagging\n",
+ "from sklearn.pipeline import make_pipeline\n",
+ "from sklearn.model_selection import cross_val_score, ShuffleSplit\n",
+ "from multi_imbalance.utils.data import load_arff_dataset\n",
+ "from sklearn.preprocessing import StandardScaler\n",
+ "from sklearn.tree import tree\n",
+ "import os\n",
+ "\n",
+ "# an example of how mrbbagging can be used in sklearn pipeline\n",
+ "X, y = load_arff_dataset(f'{os.getcwd()}\\\\data\\\\arff\\\\new_ecoli.arff')\n",
+ "clf = make_pipeline(StandardScaler(), MRBBagging(30, tree.DecisionTreeClassifier()))\n",
+ "cv = ShuffleSplit(n_splits=5, test_size=0.3, random_state=0)\n",
+ "print(cross_val_score(clf, X, y, cv=cv))"
+ ]
  }
  ],
  "metadata": {
@@ -31,4 +45,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 0
-}
+}