Skip to content

Commit 79d03c5

Browse files
committed
Updated readme [skip ci]
1 parent 4214011 commit 79d03c5

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -45,13 +45,13 @@ booster.predict(dtest)
4545
Save the model to a file
4646

4747
```ruby
48-
booster.save_model("my.model")
48+
booster.save_model("model.json")
4949
```
5050

5151
Load the model from a file
5252

5353
```ruby
54-
booster = XGBoost::Booster.new(model_file: "my.model")
54+
booster = XGBoost::Booster.new(model_file: "model.json")
5555
```
5656

5757
Get the importance of features
@@ -108,13 +108,13 @@ model.predict(x)
108108
Save the model to a file
109109

110110
```ruby
111-
model.save_model("my.model")
111+
model.save_model("model.json")
112112
```
113113

114114
Load the model from a file
115115

116116
```ruby
117-
model.load_model("my.model")
117+
model.load_model("model.json")
118118
```
119119

120120
Get the importance of features

0 commit comments

Comments
 (0)