Исходный код qiskit_machine_learning.algorithms.classifiers.neural_network_classifier

# This code is part of Qiskit.
# (C) Copyright IBM 2021, 2022.
# This code is licensed under the Apache License, Version 2.0. You may
# obtain a copy of this license in the LICENSE.txt file in the root directory
# of this source tree or at
# Any modifications or derivative works of this code must retain this
# copyright notice, and modified files need to carry a notice indicating
# that they have been altered from the originals.
"""An implementation of quantum neural network classifier."""

from typing import Union, Optional, Callable, Tuple, cast

import numpy as np
import scipy.sparse
from qiskit.algorithms.optimizers import Optimizer
from scipy.sparse import spmatrix
from sklearn.base import ClassifierMixin
from sklearn.preprocessing import OneHotEncoder, LabelEncoder

from ..objective_functions import (
from ..trainable_model import TrainableModel
from ...exceptions import QiskitMachineLearningError
from ...neural_networks import NeuralNetwork
from ...utils.loss_functions import Loss

[документация]class NeuralNetworkClassifier(TrainableModel, ClassifierMixin): """Quantum neural network classifier. Implements Scikit-Learn compatible methods for classification and extends ``ClassifierMixin``. See `Scikit-Learn <>`__ for more details. """ def __init__( self, neural_network: NeuralNetwork, loss: Union[str, Loss] = "squared_error", one_hot: bool = False, optimizer: Optional[Optimizer] = None, warm_start: bool = False, initial_point: np.ndarray = None, callback: Optional[Callable[[np.ndarray, float], None]] = None, ): """ Args: neural_network: An instance of an quantum neural network. If the neural network has a one-dimensional output, i.e., `neural_network.output_shape=(1,)`, then it is expected to return values in [-1, +1] and it can only be used for binary classification. If the output is multi-dimensional, it is assumed that the result is a probability distribution, i.e., that the entries are non-negative and sum up to one. Then there are two options, either one-hot encoding or not. In case of one-hot encoding, each probability vector resulting a neural network is considered as one sample and the loss function is applied to the whole vector. Otherwise, each entry of the probability vector is considered as an individual sample and the loss function is applied to the index and weighted with the corresponding probability. loss: A target loss function to be used in training. Default is `squared_error`, i.e. L2 loss. Can be given either as a string for 'absolute_error' (i.e. L1 Loss), 'squared_error', 'cross_entropy', 'cross_entropy_sigmoid', or as a loss function implementing the Loss interface. one_hot: Determines in the case of a multi-dimensional result of the neural_network how to interpret the result. If True it is interpreted as a single one-hot-encoded sample (e.g. for 'CrossEntropy' loss function), and if False as a set of individual predictions with occurrence probabilities (the index would be the prediction and the value the corresponding frequency, e.g. for absolute/squared loss). In case of a one-dimensional categorical output, this option determines how to encode the target data (i.e. one-hot or integer encoding). optimizer: An instance of an optimizer to be used in training. When `None` defaults to SLSQP. warm_start: Use weights from previous fit to start next fit. initial_point: Initial point for the optimizer to start from. callback: a reference to a user's callback function that has two parameters and returns ``None``. The callback can access intermediate data during training. On each iteration an optimizer invokes the callback and passes current weights as an array and a computed value as a float of the objective function being optimized. This allows to track how well optimization / training process is going on. Raises: QiskitMachineLearningError: unknown loss, invalid neural network """ super().__init__(neural_network, loss, optimizer, warm_start, initial_point, callback) self._one_hot = one_hot # encodes the target data if categorical self._target_encoder = OneHotEncoder(sparse=False) if one_hot else LabelEncoder()
[документация] def fit(self, X: np.ndarray, y: np.ndarray): # pylint: disable=invalid-name if not self._warm_start: self._fit_result = None X, y = self._validate_input(X, y) # mypy definition function: ObjectiveFunction = None if self._neural_network.output_shape == (1,): if len(y.shape) != 1 or len(np.unique(y)) != 2: raise QiskitMachineLearningError( f"Current settings only applicable to binary classification! Got labels: {y}" ) function = BinaryObjectiveFunction(X, y, self._neural_network, self._loss) else: if self._one_hot: function = OneHotObjectiveFunction(X, y, self._neural_network, self._loss) else: function = MultiClassObjectiveFunction(X, y, self._neural_network, self._loss) objective = self._get_objective(function) self._fit_result = self._optimizer.minimize( fun=objective, x0=self._choose_initial_point(), jac=function.gradient, ) return self
[документация] def predict(self, X: np.ndarray) -> np.ndarray: # pylint: disable=invalid-name if self._fit_result is None: raise QiskitMachineLearningError("Model needs to be fit to some training data first!") X, _ = self._validate_input(X) if self._neural_network.output_shape == (1,): predict = np.sign(self._neural_network.forward(X, self._fit_result.x)) else: forward = self._neural_network.forward(X, self._fit_result.x) predict_ = np.argmax(forward, axis=1) if self._one_hot: predict = np.zeros(forward.shape) for i, v in enumerate(predict_): predict[i, v] = 1 else: predict = predict_ return predict
# pylint: disable=invalid-name
[документация] def score( self, X: np.ndarray, y: np.ndarray, sample_weight: Optional[np.ndarray] = None ) -> float: X, y = self._validate_input(X, y) return ClassifierMixin.score(self, X, y, sample_weight)
def _validate_input(self, X: np.ndarray, y: np.ndarray = None) -> Tuple[np.ndarray, np.ndarray]: """ Validates and transforms if required features and labels. If arrays are sparse, they are converted to dense as the numpy math in the loss/objective functions does not work with sparse. If one hot encoding is required, then labels are one hot encoded otherwise label are encoded via ``LabelEncoder`` from ``SciKit-Learn``. If labels are strings, they converted to numerical representation. Args: X: features y: labels Returns: A tuple with validated features and labels. """ if scipy.sparse.issparse(X): # our math does not work with sparse arrays X = cast(spmatrix, X).toarray() # cast is required by mypy if y is not None: if isinstance(y[0], str): # string data is assumed to be categorical # OneHotEncoder expects data with shape (n_samples, n_features) but # LabelEncoder expects shape (n_samples,) so set desired shape y = y.reshape(-1, 1) if self._one_hot else y if self._fit_result is None: # the model is being trained, fit first y = self._target_encoder.transform(y) elif scipy.sparse.issparse(y): y = cast(spmatrix, y).toarray() # cast is required by mypy return X, y

© Copyright 2018, 2022, Qiskit Machine Learning Development Team.

Built with Sphinx using a theme provided by Read the Docs.