Passed
Push — master ( fb398d...c64a78 )
by Simon
01:38 queued 11s
created

DatasetFeatures.get_features()   A

Complexity

Conditions 1

Size

Total Lines 16
Code Lines 12

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
cc 1
eloc 12
nop 1
dl 0
loc 16
rs 9.8
c 0
b 0
f 0
1
# Author: Simon Blanke
2
# Email: [email protected]
3
# License: MIT License
4
5
import numpy as np
6
import pandas as pd
7
8
9
class DatasetFeatures:
10
    def __init__(self, X, y):
11
        self.X = X
12
        self.y = y
13
14
    def get_features(self):
15
        features_X_list = [self.X.size, self.X.itemsize, self.X.ndim]
16
        features_y_list = [self.y.size, self.y.itemsize, self.y.ndim]
17
18
        features_np = np.array(features_X_list + features_y_list)
19
20
        col_names = [
21
            "X_size",
22
            "X_byte_size",
23
            "X_ndim",
24
            "y_size",
25
            "y_byte_size",
26
            "y_ndim",
27
        ]
28
29
        return pd.DataFrame(features_np, columns=col_names)
30