Dodanie obsługi parametru epochs do lab 06 zad 1
This commit is contained in:
parent
6462a5fce5
commit
d04500d1d2
@ -1,5 +1,5 @@
|
|||||||
## Klasyfikacja jakości diamentu
|
## Klasyfikacja jakości diamentu
|
||||||
import os
|
import argparse
|
||||||
import pandas as pd
|
import pandas as pd
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import pickle
|
import pickle
|
||||||
@ -10,8 +10,11 @@ from tensorflow.keras.callbacks import History
|
|||||||
from sklearn.preprocessing import LabelEncoder, StandardScaler, OneHotEncoder
|
from sklearn.preprocessing import LabelEncoder, StandardScaler, OneHotEncoder
|
||||||
from tensorflow.keras.utils import to_categorical
|
from tensorflow.keras.utils import to_categorical
|
||||||
|
|
||||||
#Wyświetlenie zbioru danych
|
# Wczytanie parametru epochs
|
||||||
epochs = int(os.environ.get('EPOCHS', 10))
|
parser = argparse.ArgumentParser()
|
||||||
|
parser.add_argument("--epochs", type=int, default=10, help="Number of epochs")
|
||||||
|
args = parser.parse_args()
|
||||||
|
epochs = args.epochs
|
||||||
|
|
||||||
# Wczytanie danych
|
# Wczytanie danych
|
||||||
data_train = pd.read_csv('dane/diamonds_train.csv')
|
data_train = pd.read_csv('dane/diamonds_train.csv')
|
||||||
|
Loading…
Reference in New Issue
Block a user