diff --git a/examples/classification_example.ipynb b/examples/classification_example.ipynb index 55a398c..83b7aa9 100644 --- a/examples/classification_example.ipynb +++ b/examples/classification_example.ipynb @@ -2,7 +2,7 @@ "cells": [ { "cell_type": "code", - "execution_count": 1, + "execution_count": null, "metadata": {}, "outputs": [], "source": [ @@ -293,7 +293,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.10.11" + "version": "3.11.9" } }, "nbformat": 4, diff --git a/src/featureranker/rankers.py b/src/featureranker/rankers.py index 9b7c1ce..fd78f85 100644 --- a/src/featureranker/rankers.py +++ b/src/featureranker/rankers.py @@ -15,7 +15,7 @@ from sklearn.svm import l1_min_c from xgboost import XGBClassifier, XGBRegressor -from .utils import ( +from featureranker.utils import ( classification_hyper_param_search, regression_hyper_param_search, ) diff --git a/src/featureranker/utils.py b/src/featureranker/utils.py index 5b3fb9c..511a299 100644 --- a/src/featureranker/utils.py +++ b/src/featureranker/utils.py @@ -10,7 +10,7 @@ from scipy.stats import spearmanr from skopt import BayesSearchCV -from .plots import plot_confusion_matrix, plot_correlations +from featureranker.plots import plot_confusion_matrix, plot_correlations model_params = {