Wrapper around LearnKit to easy its integration in Swift CocoaPods.
HRLAlgorithms is available through CocoaPods. To install it, simply add the following line to your Podfile:
pod "HRLAlgorithms"
#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]);
HRLAlgorithms is available under the MIT license. See the LICENSE file for more info.