README.md in svmkit-0.1.0 vs README.md in svmkit-0.1.1
- old
+ new
@@ -34,10 +34,10 @@
transformer = SVMKit::KernelApproximation::RBF.new(gamma: 2.0, n_components: 1024, random_seed: 1)
transformed = transformer.fit_transform(normalized)
base_classifier =
- SVMKit::LinearModel::PegasosSVC.new(penalty: 1.0, max_iter: 50, batch_size: 20, random_seed: 1)
+ SVMKit::LinearModel::PegasosSVC.new(reg_param: 1.0, max_iter: 50, batch_size: 20, random_seed: 1)
classifier = SVMKit::Multiclass::OneVsRestClassifier.new(estimator: base_classifier)
classifier.fit(transformed, labels)
File.open('trained_normalizer.dat', 'wb') { |f| f.write(Marshal.dump(normalizer)) }
File.open('trained_transformer.dat', 'wb') { |f| f.write(Marshal.dump(transformer)) }