32bytes/plot.py at main · MLFinalProject/32bytes · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

from data_insight import DataInsight
from dataset import Dataset
import pandas as pd

data_insight = DataInsight()
# dataset = Dataset()
# train_test_df = pd.get_dummies(dataset.train_test_df)
# print(dataset.train_test_df)
# print(dataset.get_train_dataset())

# Test 1/8
# print(data_insight.compare_train_test('country'))
# data_insight.plot_train_unique('country')
# exit()

print(data_insight.train_df.arrival_date_day_of_month)

attribute_list = list(data_insight.train_df.columns.values)[:-2]
for attribute in attribute_list:
    data_insight.plot_test(attribute)
    # data_insight.plot_attribute_adr(attribute)
    # data_insight.plot_attribute_adr_mean(attribute)
    # data_insight.plot_attribute_is_canceled_mean(attribute)


# data_insight.plot_attribute_adr_mean("lead_time")