|
import pandas as pd |
|
import numpy as np |
|
|
|
|
|
num_samples = 1000 |
|
|
|
|
|
gpu = np.random.randint(1, 11, size=num_samples) |
|
ram = np.random.randint(4, 33, size=num_samples) |
|
processor = np.random.randint(1, 9, size=num_samples) |
|
|
|
|
|
|
|
def is_good_for_transformer(gpu, ram, processor): |
|
return ((gpu >= 6) & (ram >= 16) & (processor >= 4)).astype(int) |
|
|
|
output = is_good_for_transformer(gpu, ram, processor) |
|
|
|
|
|
data = pd.DataFrame({'GPU': gpu, 'RAM': ram, 'Processor': processor, 'Good_for_Transformer': output}) |
|
|
|
|
|
data.to_csv(r'Data_csv\transformer_dataset.csv', index=False) |