SYMBOL INDEX (420 symbols across 47 files) FILE: examples/MIG-Support/device-plugins/gpu-mig/src/main/java/com/nvidia/spark/NvidiaGPUMigPluginForRuntimeV2.java class NvidiaGPUMigPluginForRuntimeV2 (line 52) | public class NvidiaGPUMigPluginForRuntimeV2 implements DevicePlugin, method getRegisterRequestInfo (line 97) | @Override method getDevices (line 103) | @Override method shouldThrowOnMultipleGPUs (line 202) | private Boolean shouldThrowOnMultipleGPUs() { method onDevicesAllocated (line 210) | @Override method onDevicesReleased (line 243) | @Override method getMajorNumber (line 249) | private String getMajorNumber(String devName) { method allocateDevices (line 268) | @Override method basicSchedule (line 282) | public void basicSchedule(Set allocation, int count, class NvidiaCommandExecutor (line 303) | public class NvidiaCommandExecutor { method getDeviceInfo (line 305) | public String getDeviceInfo() throws IOException { method getDeviceMigInfo (line 311) | public String getDeviceMigInfo() throws IOException { method getMajorMinorInfo (line 316) | public String getMajorMinorInfo(String devName) throws IOException { method searchBinary (line 324) | public void searchBinary() throws Exception { method setPathOfGpuBinary (line 363) | public void setPathOfGpuBinary(String pOfGpuBinary) { method setShellExecutor (line 368) | public void setShellExecutor(NvidiaCommandExecutor shellExecutor) { method setMigDevices (line 373) | public void setMigDevices(Map migDevices) { method setShouldThrowOnMultipleGPUFromConf (line 378) | public void setShouldThrowOnMultipleGPUFromConf(Boolean shouldThrow) { FILE: examples/MIG-Support/device-plugins/gpu-mig/src/test/java/com/nvidia/spark/TestNvidiaGPUMigPluginForRuntimeV2.java class TestNvidiaGPUMigPluginForRuntimeV2 (line 37) | public class TestNvidiaGPUMigPluginForRuntimeV2 { method testGetNvidiaDevices (line 42) | @Test method testOnDeviceAllocatedMultiGPU (line 93) | @Test(expected = Exception.class) method testMultiGPUsEnvPrecedence (line 125) | @Test method testMultiGPUsConf (line 162) | @Test method testOnDeviceAllocatedMig (line 195) | @Test method testOnDeviceAllocatedNoMig (line 221) | @Test FILE: examples/ML+DL-Examples/Spark-DL/dl_inference/server_utils.py function _find_ports (line 41) | def _find_ports(num_ports: int, start_port: int = 7000) -> List[int]: function _get_valid_vllm_parameters_task (line 55) | def _get_valid_vllm_parameters_task() -> Set[str]: function _start_triton_server_task (line 74) | def _start_triton_server_task( function _start_vllm_server_task (line 158) | def _start_vllm_server_task( function _stop_server_task (line 225) | def _stop_server_task( class ServerManager (line 255) | class ServerManager: method __init__ (line 270) | def __init__(self, model_name: str, model_path: Optional[str] = None): method _get_num_executors (line 284) | def _get_num_executors(self) -> int: method host_to_http_url (line 299) | def host_to_http_url(self) -> Dict[str, str]: method _get_node_rdd (line 310) | def _get_node_rdd(self) -> RDD: method _use_stage_level_scheduling (line 316) | def _use_stage_level_scheduling(self, rdd: RDD) -> RDD: method start_servers (line 354) | def start_servers( method stop_servers (line 399) | def stop_servers( class TritonServerManager (line 443) | class TritonServerManager(ServerManager): method __init__ (line 459) | def __init__(self, model_name: str, model_path: Optional[str] = None): method host_to_grpc_url (line 463) | def host_to_grpc_url(self) -> Dict[str, str]: method start_servers (line 474) | def start_servers( class VLLMServerManager (line 499) | class VLLMServerManager(ServerManager): method __init__ (line 517) | def __init__(self, model_name: str, model_path: str = None): method _get_valid_vllm_parameters (line 521) | def _get_valid_vllm_parameters(self) -> List[str]: method _validate_vllm_kwargs (line 526) | def _validate_vllm_kwargs(self, kwargs: Dict[str, Any]): method start_servers (line 545) | def start_servers( FILE: examples/SQL+DF-Examples/customer-churn/notebooks/python/churn/augment.py function get_currency_type (line 33) | def get_currency_type(): function _register_session (line 52) | def _register_session(s): function _get_uniques (line 56) | def _get_uniques(ct): function register_options (line 95) | def register_options(**kwargs): function load_supplied_data (line 100) | def load_supplied_data(session, input_file): function replicate_df (line 150) | def replicate_df(df, duplicates): function examine_categoricals (line 163) | def examine_categoricals(df, columns=None): function billing_events (line 189) | def billing_events(df): function resolve_path (line 244) | def resolve_path(name): function write_df (line 254) | def write_df(df, name, skip_replication=False, partition_by=None): function customer_meta (line 275) | def customer_meta(df): function phone_features (line 330) | def phone_features(df): function internet_features (line 341) | def internet_features(df): function account_features (line 370) | def account_features(df): function debug_augmentation (line 393) | def debug_augmentation(df): FILE: examples/SQL+DF-Examples/customer-churn/notebooks/python/churn/eda.py function isnumeric (line 20) | def isnumeric(data_type): function percent_true (line 25) | def percent_true(df, cols): function cardinalities (line 30) | def cardinalities(df, cols): function likely_unique (line 40) | def likely_unique(counts): function likely_categoricals (line 45) | def likely_categoricals(counts): function unique_values (line 49) | def unique_values(df, cols): function unique_values_array (line 55) | def unique_values_array(df, cols): function unique_values_driver (line 69) | def unique_values_driver(df, cols): function approx_ecdf (line 72) | def approx_ecdf(df, cols): function gen_summary (line 83) | def gen_summary(df, output_prefix=""): function losses_by_month (line 125) | def losses_by_month(be): function output_reports (line 129) | def output_reports(df, be=None, report_prefix=""): FILE: examples/SQL+DF-Examples/customer-churn/notebooks/python/churn/etl.py function register_options (line 29) | def register_options(**kwargs): function _register_session (line 34) | def _register_session(s): function _register_views (line 38) | def _register_views(lvars, *names): function withsession (line 43) | def withsession(df_arg=0): function read_df (line 51) | def read_df(session, fn): function find_customers (line 61) | def find_customers(billing_events_df): function customers (line 68) | def customers(): function join_billing_data (line 72) | def join_billing_data(billing_events_df): function join_phone_features (line 117) | def join_phone_features(phone_features_df): function untidy_feature (line 152) | def untidy_feature(df, feature): function chained_join (line 158) | def chained_join(column, base_df, dfs, how="leftouter"): function resolve_nullable_column (line 166) | def resolve_nullable_column(df, col, null_val="No"): function resolve_dependent_column (line 170) | def resolve_dependent_column( function join_internet_features (line 184) | def join_internet_features(internet_features_df): function join_account_features (line 247) | def join_account_features(account_features_df): function process_account_meta (line 271) | def process_account_meta(account_meta_df, usecal=None): function forcefloat (line 302) | def forcefloat(c): function join_wide_table (line 306) | def join_wide_table(customer_billing, customer_phone_features, customer_... function cast_and_coalesce_wide_data (line 346) | def cast_and_coalesce_wide_data(wd): function write_df (line 374) | def write_df(df, name): FILE: examples/UDF-Examples/RAPIDS-accelerated-UDFs/conftest.py function pytest_addoption (line 15) | def pytest_addoption(parser): FILE: examples/UDF-Examples/RAPIDS-accelerated-UDFs/src/main/cpp/benchmarks/cosine_similarity/cosine_similarity_benchmark.cpp function cosine_similarity_bench_args (line 26) | static void cosine_similarity_bench_args(benchmark::internal::Benchmark* b) function BM_cosine_similarity (line 45) | static void BM_cosine_similarity(benchmark::State& state) class CosineSimilarity (line 82) | class CosineSimilarity : public native_udf::benchmark { FILE: examples/UDF-Examples/RAPIDS-accelerated-UDFs/src/main/cpp/benchmarks/fixture/benchmark_fixture.hpp type native_udf (line 23) | namespace native_udf { function make_cuda (line 27) | inline auto make_cuda() { return std::make_shared v) { method evaluateColumnar (line 63) | @Override method cosineSimilarity (line 81) | private static native long cosineSimilarity(long vectorView1, long vec... FILE: examples/UDF-Examples/RAPIDS-accelerated-UDFs/src/main/java/com/nvidia/spark/rapids/udf/java/DecimalFraction.java class DecimalFraction (line 31) | public class DecimalFraction implements UDF1, Ra... method call (line 33) | @Override method evaluateColumnar (line 42) | @Override FILE: examples/UDF-Examples/RAPIDS-accelerated-UDFs/src/main/java/com/nvidia/spark/rapids/udf/java/NativeUDFExamplesLoader.java class NativeUDFExamplesLoader (line 24) | public class NativeUDFExamplesLoader { method ensureLoaded (line 28) | public static synchronized void ensureLoaded() { FILE: examples/UDF-Examples/RAPIDS-accelerated-UDFs/src/main/java/com/nvidia/spark/rapids/udf/java/URLDecode.java class URLDecode (line 34) | public class URLDecode implements UDF1, RapidsUDF { method call (line 36) | @Override method evaluateColumnar (line 53) | @Override FILE: examples/UDF-Examples/RAPIDS-accelerated-UDFs/src/main/java/com/nvidia/spark/rapids/udf/java/URLEncode.java class URLEncode (line 33) | public class URLEncode implements UDF1, RapidsUDF { method call (line 35) | @Override method evaluateColumnar (line 52) | @Override FILE: examples/UDF-Examples/RAPIDS-accelerated-UDFs/src/main/python/asserts.py function _assert_equal (line 28) | def _assert_equal(cpu, gpu, float_check, path): function assert_equal (line 100) | def assert_equal(cpu, gpu): function _has_incompat_conf (line 109) | def _has_incompat_conf(conf): class _RowCmp (line 113) | class _RowCmp(object): method __init__ (line 115) | def __init__(self, wrapped): method cmp (line 132) | def cmp(self, other): method __lt__ (line 159) | def __lt__(self, other): method __gt__ (line 162) | def __gt__(self, other): method __eq__ (line 165) | def __eq__(self, other): method __le__ (line 168) | def __le__(self, other): method __ge__ (line 171) | def __ge__(self, other): method __ne__ (line 174) | def __ne__(self, other): function _prep_func_for_compare (line 177) | def _prep_func_for_compare(func, mode): function _prep_incompat_conf (line 216) | def _prep_incompat_conf(conf): function _assert_gpu_and_cpu_writes_are_equal (line 224) | def _assert_gpu_and_cpu_writes_are_equal( function assert_gpu_and_cpu_writes_are_equal_collect (line 258) | def assert_gpu_and_cpu_writes_are_equal_collect(write_func, read_func, b... function assert_gpu_and_cpu_writes_are_equal_iterator (line 267) | def assert_gpu_and_cpu_writes_are_equal_iterator(write_func, read_func, ... function assert_gpu_fallback_write (line 276) | def assert_gpu_fallback_write(write_func, function assert_cpu_and_gpu_are_equal_collect_with_capture (line 312) | def assert_cpu_and_gpu_are_equal_collect_with_capture(func, function assert_cpu_and_gpu_are_equal_sql_with_capture (line 343) | def assert_cpu_and_gpu_are_equal_sql_with_capture(df_fun, function assert_gpu_fallback_collect (line 361) | def assert_gpu_fallback_collect(func, function assert_gpu_sql_fallback_collect (line 386) | def assert_gpu_sql_fallback_collect(df_fun, cpu_fallback_class_name, tab... function _assert_gpu_and_cpu_are_equal (line 398) | def _assert_gpu_and_cpu_are_equal(func, function run_with_cpu (line 438) | def run_with_cpu(func, function run_with_cpu_and_gpu (line 464) | def run_with_cpu_and_gpu(func, function assert_gpu_and_cpu_are_equal_collect (line 499) | def assert_gpu_and_cpu_are_equal_collect(func, conf={}, is_cpu_first=True): function assert_gpu_and_cpu_are_equal_iterator (line 507) | def assert_gpu_and_cpu_are_equal_iterator(func, conf={}, is_cpu_first=Tr... function assert_gpu_and_cpu_row_counts_equal (line 515) | def assert_gpu_and_cpu_row_counts_equal(func, conf={}, is_cpu_first=True): function assert_gpu_and_cpu_are_equal_sql (line 523) | def assert_gpu_and_cpu_are_equal_sql(df_fun, table_name, sql, conf=None,... function assert_py4j_exception (line 549) | def assert_py4j_exception(func, error_message): function assert_gpu_and_cpu_error (line 560) | def assert_gpu_and_cpu_error(df_fun, conf, error_message): function with_cpu_sql (line 572) | def with_cpu_sql(df_fun, table_name, sql, conf=None, debug=False): FILE: examples/UDF-Examples/RAPIDS-accelerated-UDFs/src/main/python/conftest.py function get_float_check (line 22) | def get_float_check(): function is_incompat (line 30) | def is_incompat(): function should_sort_on_spark (line 36) | def should_sort_on_spark(): function should_sort_locally (line 39) | def should_sort_locally(): function is_allowing_any_non_gpu (line 45) | def is_allowing_any_non_gpu(): function get_non_gpu_allowed (line 48) | def get_non_gpu_allowed(): function get_validate_execs_in_gpu_plan (line 51) | def get_validate_execs_in_gpu_plan(): function runtime_env (line 56) | def runtime_env(): function is_apache_runtime (line 59) | def is_apache_runtime(): function is_databricks_runtime (line 62) | def is_databricks_runtime(): function is_emr_runtime (line 65) | def is_emr_runtime(): function is_dataproc_runtime (line 68) | def is_dataproc_runtime(): function is_nightly_run (line 74) | def is_nightly_run(): function is_at_least_precommit_run (line 77) | def is_at_least_precommit_run(): function skip_unless_nightly_tests (line 80) | def skip_unless_nightly_tests(description): function skip_unless_precommit_tests (line 86) | def skip_unless_precommit_tests(description): function get_limit (line 96) | def get_limit(): function _get_limit_from_mark (line 99) | def _get_limit_from_mark(mark): function pytest_runtest_setup (line 105) | def pytest_runtest_setup(item): function pytest_configure (line 182) | def pytest_configure(config): function pytest_collection_modifyitems (line 195) | def pytest_collection_modifyitems(config, items): function std_input_path (line 228) | def std_input_path(request): function spark_tmp_path (line 236) | def spark_tmp_path(request): class TmpTableFactory (line 252) | class TmpTableFactory: method __init__ (line 253) | def __init__(self, base_id): method get (line 257) | def get(self): function spark_tmp_table_factory (line 263) | def spark_tmp_table_factory(request): function _get_jvm_session (line 273) | def _get_jvm_session(spark): function _get_jvm (line 276) | def _get_jvm(spark): function spark_jvm (line 279) | def spark_jvm(): class MortgageRunner (line 282) | class MortgageRunner: method __init__ (line 283) | def __init__(self, mortgage_format, mortgage_acq_path, mortgage_perf_p... method do_test_query (line 288) | def do_test_query(self, spark): function mortgage (line 306) | def mortgage(request): function enable_cudf_udf (line 319) | def enable_cudf_udf(request): function enable_rapids_udf_example_native (line 326) | def enable_rapids_udf_example_native(request): FILE: examples/UDF-Examples/RAPIDS-accelerated-UDFs/src/main/python/data_gen.py class DataGen (line 30) | class DataGen: method __repr__ (line 33) | def __repr__(self): method __hash__ (line 38) | def __hash__(self): method __eq__ (line 41) | def __eq__(self, other): method __ne__ (line 44) | def __ne__(self, other): method __init__ (line 47) | def __init__(self, data_type, nullable=True, special_cases =[]): method copy_special_case (line 69) | def copy_special_case(self, special_case, weight=1.0): method with_special_case (line 76) | def with_special_case(self, special_case, weight=1.0): method get_types (line 91) | def get_types(self): method start (line 95) | def start(self, rand): method _start (line 99) | def _start(self, rand, gen_func): method gen (line 119) | def gen(self, force_no_nulls=False): method contains_ts (line 129) | def contains_ts(self): class ConvertGen (line 133) | class ConvertGen(DataGen): method __init__ (line 135) | def __init__(self, child_gen, func, data_type=None, nullable=True): method __repr__ (line 142) | def __repr__(self): method start (line 145) | def start(self, rand): class StringGen (line 153) | class StringGen(DataGen): method __init__ (line 155) | def __init__(self, pattern="(.|\n){1,30}", flags=0, charset=sre_yield.... method with_special_pattern (line 159) | def with_special_pattern(self, pattern, flags=0, charset=sre_yield.CHA... method start (line 171) | def start(self, rand): class ByteGen (line 181) | class ByteGen(DataGen): method __init__ (line 183) | def __init__(self, nullable=True, min_val = BYTE_MIN, max_val = BYTE_M... method start (line 188) | def start(self, rand): class ShortGen (line 193) | class ShortGen(DataGen): method __init__ (line 195) | def __init__(self, nullable=True, min_val = SHORT_MIN, max_val = SHORT... method start (line 201) | def start(self, rand): class IntegerGen (line 206) | class IntegerGen(DataGen): method __init__ (line 208) | def __init__(self, nullable=True, min_val = INT_MIN, max_val = INT_MAX, method start (line 214) | def start(self, rand): class DecimalGen (line 217) | class DecimalGen(DataGen): method __init__ (line 219) | def __init__(self, precision=None, scale=None, nullable=True, special_... method __repr__ (line 232) | def __repr__(self): method start (line 235) | def start(self, rand): class LongGen (line 245) | class LongGen(DataGen): method __init__ (line 247) | def __init__(self, nullable=True, min_val = LONG_MIN, max_val = LONG_M... method start (line 253) | def start(self, rand): class LongRangeGen (line 256) | class LongRangeGen(DataGen): method __init__ (line 258) | def __init__(self, nullable=False, start_val=0, direction="inc"): method start (line 275) | def start(self, rand): class RepeatSeqGen (line 279) | class RepeatSeqGen(DataGen): method __init__ (line 281) | def __init__(self, child, length): method __repr__ (line 289) | def __repr__(self): method _loop_values (line 292) | def _loop_values(self): method start (line 297) | def start(self, rand): class SetValuesGen (line 303) | class SetValuesGen(DataGen): method __init__ (line 305) | def __init__(self, data_type, data): method __repr__ (line 310) | def __repr__(self): method start (line 313) | def start(self, rand): class FloatGen (line 324) | class FloatGen(DataGen): method __init__ (line 326) | def __init__(self, nullable=True, method _fixup_nans (line 338) | def _fixup_nans(self, v): method start (line 343) | def start(self, rand): class DoubleGen (line 359) | class DoubleGen(DataGen): method __init__ (line 361) | def __init__(self, min_exp=DOUBLE_MIN_EXP, max_exp=DOUBLE_MAX_EXP, no_... method make_from (line 388) | def make_from(sign, exp, fraction): method _fixup_nans (line 397) | def _fixup_nans(self, v): method start (line 402) | def start(self, rand): class BooleanGen (line 417) | class BooleanGen(DataGen): method __init__ (line 419) | def __init__(self, nullable=True): method start (line 422) | def start(self, rand): class StructGen (line 425) | class StructGen(DataGen): method __init__ (line 427) | def __init__(self, children, nullable=True, special_cases=[]): method __repr__ (line 438) | def __repr__(self): method start (line 441) | def start(self, rand): method contains_ts (line 449) | def contains_ts(self): class DateGen (line 452) | class DateGen(DataGen): method __init__ (line 454) | def __init__(self, start=None, end=None, nullable=True): method _guess_leap_year (line 491) | def _guess_leap_year(t): method _to_days_since_epoch (line 501) | def _to_days_since_epoch(self, val): method _from_days_since_epoch (line 504) | def _from_days_since_epoch(self, days): method start (line 507) | def start(self, rand): class TimestampGen (line 512) | class TimestampGen(DataGen): method __init__ (line 514) | def __init__(self, start=None, end=None, nullable=True): method _to_ms_since_epoch (line 542) | def _to_ms_since_epoch(self, val): method _from_ms_since_epoch (line 545) | def _from_ms_since_epoch(self, ms): method start (line 548) | def start(self, rand): method contains_ts (line 553) | def contains_ts(self): class ArrayGen (line 556) | class ArrayGen(DataGen): method __init__ (line 558) | def __init__(self, child_gen, min_length=0, max_length=20, nullable=Tr... method __repr__ (line 565) | def __repr__(self): method start (line 568) | def start(self, rand): method contains_ts (line 577) | def contains_ts(self): class MapGen (line 580) | class MapGen(DataGen): method __init__ (line 582) | def __init__(self, key_gen, value_gen, min_length=0, max_length=20, nu... method __repr__ (line 591) | def __repr__(self): method start (line 594) | def start(self, rand): method contains_ts (line 602) | def contains_ts(self): class NullGen (line 606) | class NullGen(DataGen): method __init__ (line 608) | def __init__(self): method start (line 611) | def start(self, rand): function skip_if_not_utc (line 616) | def skip_if_not_utc(): function gen_df (line 620) | def gen_df(spark, data_gen, length=2048, seed=0, num_slices=None): function _mark_as_lit (line 642) | def _mark_as_lit(data, data_type): function _gen_scalars_common (line 675) | def _gen_scalars_common(data_gen, count, seed=0): function gen_scalars (line 689) | def gen_scalars(data_gen, count, seed=0, force_no_nulls=False): function gen_scalar (line 697) | def gen_scalar(data_gen, seed=0, force_no_nulls=False): function gen_scalar_values (line 702) | def gen_scalar_values(data_gen, count, seed=0, force_no_nulls=False): function gen_scalar_value (line 707) | def gen_scalar_value(data_gen, seed=0, force_no_nulls=False): function debug_df (line 712) | def debug_df(df, path = None, file_format = 'json', num_parts = 1): function print_params (line 737) | def print_params(data_gen): function idfn (line 740) | def idfn(val): function meta_idfn (line 744) | def meta_idfn(meta): function three_col_df (line 749) | def three_col_df(spark, a_gen, b_gen, c_gen, length=2048, seed=0, num_sl... function two_col_df (line 753) | def two_col_df(spark, a_gen, b_gen, length=2048, seed=0, num_slices=None): function binary_op_df (line 757) | def binary_op_df(spark, gen, length=2048, seed=0, num_slices=None): function unary_op_df (line 760) | def unary_op_df(spark, gen, length=2048, seed=0, num_slices=None): function to_cast_string (line 764) | def to_cast_string(spark_type): function get_null_lit_string (line 795) | def get_null_lit_string(spark_type): function _convert_to_sql (line 802) | def _convert_to_sql(spark_type, data): function gen_scalars_for_sql (line 829) | def gen_scalars_for_sql(data_gen, count, seed=0, force_no_nulls=False): function copy_and_update (line 961) | def copy_and_update(conf, *more_confs): function append_unique_int_col_to_df (line 999) | def append_unique_int_col_to_df(spark, dataframe): FILE: examples/UDF-Examples/RAPIDS-accelerated-UDFs/src/main/python/rapids_udf_test.py function drop_udf (line 24) | def drop_udf(spark, udfname): function skip_if_no_hive (line 27) | def skip_if_no_hive(spark): function load_hive_udf_or_skip_test (line 31) | def load_hive_udf_or_skip_test(spark, udfname, udfclass): function test_hive_simple_udf (line 35) | def test_hive_simple_udf(): function test_hive_generic_udf (line 46) | def test_hive_generic_udf(): function test_hive_simple_udf_native (line 65) | def test_hive_simple_udf_native(): function load_java_udf_or_skip_test (line 76) | def load_java_udf_or_skip_test(spark, udfname, udfclass, udf_return_type... function test_java_url_decode (line 80) | def test_java_url_decode(): function test_java_url_encode (line 86) | def test_java_url_encode(): function test_java_decimal_fraction (line 92) | def test_java_decimal_fraction(): function test_java_cosine_similarity_reasonable_range (line 108) | def test_java_cosine_similarity_reasonable_range(): function test_java_cosine_similarity_with_nans (line 120) | def test_java_cosine_similarity_with_nans(): FILE: examples/UDF-Examples/RAPIDS-accelerated-UDFs/src/main/python/spark_init_internal.py function _spark__init (line 26) | def _spark__init(): function _handle_derby_dir (line 59) | def _handle_derby_dir(sb, driver_opts, wid): function _handle_event_log_dir (line 66) | def _handle_event_log_dir(sb, wid): function get_spark_i_know_what_i_am_doing (line 100) | def get_spark_i_know_what_i_am_doing(): function spark_version (line 109) | def spark_version(): FILE: examples/UDF-Examples/RAPIDS-accelerated-UDFs/src/main/python/spark_session.py function _from_scala_map (line 20) | def _from_scala_map(scala_map): function is_tz_utc (line 34) | def is_tz_utc(spark=_spark): function _set_all_confs (line 44) | def _set_all_confs(conf): function reset_spark_session_conf (line 49) | def reset_spark_session_conf(): function _check_for_proper_return_values (line 60) | def _check_for_proper_return_values(something): function with_spark_session (line 65) | def with_spark_session(func, conf={}): function _add_job_description (line 75) | def _add_job_description(conf): function with_cpu_session (line 82) | def with_cpu_session(func, conf={}): function with_gpu_session (line 88) | def with_gpu_session(func, conf={}): function is_before_spark_311 (line 105) | def is_before_spark_311(): function is_before_spark_320 (line 108) | def is_before_spark_320(): function is_before_spark_330 (line 111) | def is_before_spark_330(): function is_databricks91_or_later (line 114) | def is_databricks91_or_later(): FILE: examples/XGBoost-Examples/agaricus/python/com/nvidia/spark/examples/agaricus/main.py function main (line 28) | def main(args, xgboost_args): FILE: examples/XGBoost-Examples/mortgage/python/com/nvidia/spark/examples/mortgage/cross_validator_main.py function main (line 25) | def main(args, xgboost_args): FILE: examples/XGBoost-Examples/mortgage/python/com/nvidia/spark/examples/mortgage/etl.py function load_data (line 25) | def load_data(spark, paths, schema, args, extra_csv_opts={}): function prepare_rawDf (line 40) | def prepare_rawDf(spark, args): function extract_perf_columns (line 50) | def extract_perf_columns(rawDf): function prepare_performance (line 87) | def prepare_performance(spark, args, rawDf): function extract_acq_columns (line 179) | def extract_acq_columns(rawDf): function prepare_acquisition (line 213) | def prepare_acquisition(spark, args, rawDf): function extract_paths (line 218) | def extract_paths(paths, prefix): function etl (line 226) | def etl(spark, args): FILE: examples/XGBoost-Examples/mortgage/python/com/nvidia/spark/examples/mortgage/etl_main.py function main (line 21) | def main(args, xgboost_args): FILE: examples/XGBoost-Examples/mortgage/python/com/nvidia/spark/examples/mortgage/main.py function main (line 24) | def main(args, xgboost_args): FILE: examples/XGBoost-Examples/taxi/python/com/nvidia/spark/examples/taxi/cross_validator_main.py function main (line 24) | def main(args, xgboost_args): FILE: examples/XGBoost-Examples/taxi/python/com/nvidia/spark/examples/taxi/etl_main.py function main (line 22) | def main(args, xgboost_args): FILE: examples/XGBoost-Examples/taxi/python/com/nvidia/spark/examples/taxi/main.py function main (line 23) | def main(args, xgboost_args): FILE: examples/XGBoost-Examples/taxi/python/com/nvidia/spark/examples/taxi/pre_process.py function pre_process (line 23) | def pre_process(data_frame): function drop_useless (line 36) | def drop_useless(data_frame): function encode_categories (line 46) | def encode_categories(data_frame): function fill_na (line 52) | def fill_na(data_frame): function remove_invalid (line 55) | def remove_invalid(data_frame): function convert_datetime (line 68) | def convert_datetime(data_frame): function add_h_distance (line 82) | def add_h_distance(data_frame): FILE: examples/XGBoost-Examples/utility/python/com/nvidia/spark/examples/main.py function main (line 20) | def main(): FILE: examples/XGBoost-Examples/utility/python/com/nvidia/spark/examples/utility/args.py function _to_bool (line 23) | def _to_bool(literal): function _to_ratio_pair (line 27) | def _to_ratio_pair(literal): # e.g., '80:20' function _validate_args (line 44) | def _validate_args(args): function _attach_derived_args (line 62) | def _attach_derived_args(args): function _inspect_xgb_parameters (line 69) | def _inspect_xgb_parameters() -> typing.Dict[str, type]: function parse_arguments (line 91) | def parse_arguments(): FILE: examples/XGBoost-Examples/utility/python/com/nvidia/spark/examples/utility/utils.py function merge_dicts (line 27) | def merge_dicts(dict_x, dict_y): function show_sample (line 33) | def show_sample(args, data_frame, label): function vectorize_data_frame (line 38) | def vectorize_data_frame(data_frame, label): function vectorize_data_frames (line 48) | def vectorize_data_frames(data_frames, label): function with_benchmark (line 52) | def with_benchmark(phrase, action): function check_classification_accuracy (line 61) | def check_classification_accuracy(data_frame, label): function check_regression_accuracy (line 69) | def check_regression_accuracy(data_frame, label): function prepare_data (line 77) | def prepare_data(spark, args, schema, dataPath): function extract_paths (line 86) | def extract_paths(paths, prefix): function transform_data (line 91) | def transform_data( function valid_input_data (line 104) | def valid_input_data(spark, args, raw_schema, final_schema): FILE: examples/spark-connect-gpu/client/python/batch-job.py function explain (line 33) | def explain(dataframe): FILE: scripts/csp-startup-scripts/emr/emr-spark-plugin-startup.py function upload_file_to_s3 (line 25) | def upload_file_to_s3(file_name, bucket_name, object_name=None): function create_emr_cluster (line 56) | def create_emr_cluster(release_label, key_name, service_role, subnet_id,... FILE: scripts/encoding/python/com/nvidia/spark/encoding/criteo/common.py function customize_reader (line 17) | def customize_reader(reader): function customize_writer (line 21) | def customize_writer(writer): FILE: scripts/encoding/python/com/nvidia/spark/encoding/criteo/one_hot_cpu_main.py function index (line 22) | def index(df, column): function expand (line 28) | def expand(indexer, df, column): function main (line 37) | def main(args): FILE: scripts/encoding/python/com/nvidia/spark/encoding/criteo/target_cpu_main.py function get_dict_df (line 25) | def get_dict_df(train_df, target_col, label_col): function encode_df (line 32) | def encode_df(original_df, dict_df, col_name): function main (line 39) | def main(args): FILE: scripts/encoding/python/com/nvidia/spark/encoding/main.py function main (line 19) | def main(): FILE: scripts/encoding/python/com/nvidia/spark/encoding/utility/args.py function _to_bool (line 21) | def _to_bool(literal): function _to_str_list (line 24) | def _to_str_list(literal): function _validate_args (line 32) | def _validate_args(args): function parse_arguments (line 48) | def parse_arguments(): FILE: scripts/encoding/python/com/nvidia/spark/encoding/utility/utils.py function load_data (line 18) | def load_data(spark, paths, args, customize=None): function save_data (line 25) | def save_data(data_frame, path, args, customize=None): function load_model (line 33) | def load_model(model_class, path): function load_models (line 36) | def load_models(model_class, paths): function save_model (line 39) | def save_model(model, path, args): function save_dict (line 43) | def save_dict(mean_dict, target_path): function load_dict (line 50) | def load_dict(dict_path): function load_dict_df (line 57) | def load_dict_df(spark, dict_df_path):