import torch import mindsdb from tests import basic_test if __name__ == "__main__": use_gpu_settings = [False] if torch.cuda.is_available(): use_gpu_settings.append(True) # Cycle through a few options: for backend in ['lightwood']: for use_gpu in use_gpu_settings: print(f'use_gpu is set to {use_gpu}, backend is set to {backend}') basic_test(backend=backend, use_gpu=use_gpu, IS_CI_TEST=True) print( '\n\n=============[Success]==============\n Finished running full test suite !\n=============[Success]==============\n\n' )
from tests import basic_test # Test with a few basic options basic_test(backend='lightwood',use_gpu=False,ignore_columns=[]) print('\n\n=============[Success]==============\n Finished running quick test !\n=============[Success]==============\n\n')
from tests import basic_test # Cycle through a few options: for backend in ['lightwood', 'ludwig']: for use_gpu in [False, True]: print(f'use_gpu is set to {use_gpu}, backend is set to {backend}') basic_test(backend=backend, use_gpu=use_gpu, ignore_columns=[]) # Try ignoring some columns basic_test(backend='lightwood', use_gpu=True, ignore_columns=['days_on_market', 'number_of_bathrooms']) print( '\n\n=============[Success]==============\n Finished running full test suite !\n=============[Success]==============\n\n' )
import torch import mindsdb from tests import basic_test if __name__ == "__main__": use_gpu_settings = [] if torch.cuda.is_available(): use_gpu_settings.append(True) use_gpu_settings.append(False) # Try ignoring some columns and running only the stats generator run_extra = True # Cycle through a few options: for backend in ['lightwood']: #,'ludwig' for use_gpu in use_gpu_settings: print(f'use_gpu is set to {use_gpu}, backend is set to {backend}') if run_extra: basic_test(backend=backend,use_gpu=use_gpu,ignore_columns=['days_on_market','number_of_bathrooms'], IS_CI_TEST=True, run_extra=True) run_extra = False else: basic_test(backend=backend,use_gpu=use_gpu,ignore_columns=[], IS_CI_TEST=True) print('\n\n=============[Success]==============\n Finished running full test suite !\n=============[Success]==============\n\n')
import torch import mindsdb from tests import basic_test if __name__ == "__main__": use_gpu_settings = [] if torch.cuda.is_available(): use_gpu_settings.append(True) use_gpu_settings.append(False) # Cycle through a few options: for backend in ['lightwood','ludwig']: for use_gpu in use_gpu_settings: print(f'use_gpu is set to {use_gpu}, backend is set to {backend}') basic_test(backend=backend,use_gpu=use_gpu,ignore_columns=[], IS_CI_TEST=True) # Try ignoring some columns and running only the stats generator basic_test(backend='lightwood',use_gpu=use_gpu_settings[0],ignore_columns=['days_on_market','number_of_bathrooms'],run_extra=True, IS_CI_TEST=True) print('\n\n=============[Success]==============\n Finished running full test suite !\n=============[Success]==============\n\n')
from tests import basic_test import torch # Test with a few basic options if __name__ == "__main__": basic_test(backend='lightwood',use_gpu=torch.cuda.is_available(),ignore_columns=[], IS_CI_TEST=True) print('\n\n=============[Success]==============\n Finished running quick test !\n=============[Success]==============\n\n')