Skip to content

README.md

Latest commit

 

History

History
50 lines (34 loc) · 1.9 KB

README.md

File metadata and controls

50 lines (34 loc) · 1.9 KB

HRLAlgorithms

CI Status codecov.io Version Docs

Wrapper around LearnKit to easy its integration in Swift CocoaPods.

Installation

HRLAlgorithms is available through CocoaPods. To install it, simply add the following line to your Podfile:

pod "HRLAlgorithms"

Usage

#import "HRLMatrix.h"
#import "HRLMatrixSplitter.h"
#import "HRLSplittedMatrix.h"
#import "HRLTrainedKNNClassifier.h"
#import "HRLTrainedKNNClassifierFactory.h"
#import "HRLVector.h"

id<HRLMatrix> matrix = ...;

id<HRLMatrixSplitter> splitter = [[HRLMatrixSplitter alloc] init];
HRLSplittedMatrix *splittedMatrix = [splitter splittedMatrixWithMatrix:matrix
                                                          trainingBias:0.75f];

id<HRLTrainedKNNClassifierFactory> factory = [[HRLTrainedKNNClassifierFactory alloc] init];
id<HRLTrainedKNNClassifier> classifier = [factory makeTrainedKNNClassifierWithMatrix:splittedMatrix.trainingMatrix
                                                                      neighborsCount:5];

NSLog(@"Accuracy: %f", [classifier estimatedAccuracyWithMatrix:splittedMatrix.testMatrix]);

id<HRLVector> vector = ...;

NSLog(@"Predicted class: %lu", (unsigned long)[classifier predictedClassForVector:vector]);

License

HRLAlgorithms is available under the MIT license. See the LICENSE file for more info.