README.md in lightgbm-0.1.3 vs README.md in lightgbm-0.1.4

- old
+ new

@@ -16,11 +16,11 @@ gem 'lightgbm' ``` ## Getting Started -This library follows the [Data Structure, Training, and Scikit-Learn APIs](https://lightgbm.readthedocs.io/en/latest/Python-API.html) of the Python library. A few differences are: +This library follows the [Python API](https://lightgbm.readthedocs.io/en/latest/Python-API.html). A few differences are: - The `get_` prefix is removed from methods - The default verbosity is `-1` - With the `cv` method, `stratified` is set to `false` @@ -61,11 +61,11 @@ ``` Early stopping ```ruby -LightGBM.train(params, train_set, valid_set: [train_set, test_set], early_stopping_rounds: 5) +LightGBM.train(params, train_set, valid_sets: [train_set, test_set], early_stopping_rounds: 5) ``` CV ```ruby @@ -112,9 +112,15 @@ Get the importance of features ```ruby model.feature_importances +``` + +Early stopping + +```ruby +model.fit(x, y, eval_set: [[x_test, y_test]], early_stopping_rounds: 5) ``` ## Data Data can be an array of arrays