5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
# File 'lib/tensorflow/keras/datasets/boston_housing.rb', line 5
def self.load_data(path: "boston_housing.npz", test_split: 0.2, seed: 113)
file = Utils.get_file(
path,
"https://storage.googleapis.com/tensorflow/tf-keras-datasets/boston_housing.npz",
file_hash: "f553886a1f8d56431e820c5b82552d9d95cfcb96d1e678153f8839538947dff5"
)
data = Npy.load_npz(file)
x = data["x"]
y = data["y"]
len = x.shape[0]
indices = (0...len).to_a.shuffle(random: Random.new(seed))
x = x[indices, true]
y = y[indices]
x_train = x[0...(len * (1 - test_split)), true]
y_train = y[0...(len * (1 - test_split))]
x_test = x[(len * (1 - test_split))..-1, true]
y_test = y[(len * (1 - test_split))..-1]
[[x_train, y_train], [x_test, y_test]]
end
|