HRLAlgorithms
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.