This package provides tools for calculating weights for test data based on training data using nearest neighbors and various distance measures. The implementation leverages scikit-learn, scipy, and joblib for efficient computation.
https://pypi.org/project/db-weights/
- Calculate weights using Nearest Neighbors
- Calculate weights using different distance measures
- Supports various distance metrics including Euclidean, Mahalanobis, Cosine, etc.
- Parallel processing for faster computation
You can install the package using pip:
pip install db-weightsImporting the package
importnumpyasnpfromdb_weightsimportWeightCalculatorCreating a WeightCalculator instance
weight_calculator=WeightCalculator(n_neighbors=3, algorithm='auto', n_jobs=-1)Calculating weights using Nearest Neighbors
x_train=np.random.rand(100, 5) # Training datax_test=np.random.rand(20, 5) # Test dataweights_nn=weight_calculator.calculate_weights_nn(x_train, x_test, weight=1)
print(weights_nn)Calculating weights using distance measures
weights_dist=weight_calculator.calculate_weights_dist(x_train, x_test, weight=1, measure_type='euclidean')
print(weights_dist)Getting available distance measures
measurements=weight_calculator.get_measurements()
print(measurements)Make WHL
pipinstallsetuptoolswheelpythonsetup.pysdistbdist_wheel- numpy
- scikit-learn
- scipy
- joblib
Contributions are welcome. Please fork the repository and submit a pull request.
This project is licensed under the MIT License - see the LICENSE file for details.