2 implementations of ComponentCatalog
Microsoft.ML.Core (1)
Environment\HostEnvironmentBase.cs (1)
359public ComponentCatalog ComponentCatalog { get; }
Microsoft.ML.Data (1)
MLContext.cs (1)
87public ComponentCatalog ComponentCatalog => _env.ComponentCatalog;
71 references to ComponentCatalog
Microsoft.ML.Core (9)
CommandLine\CmdParser.cs (3)
658_host.ComponentCatalog.TryFindComponentCaseInsensitive(arg.Field.FieldType, value, out component)) 1926var catalog = env.ComponentCatalog; 2109var catalog = env.ComponentCatalog;
ComponentModel\AssemblyLoadingUtils.cs (3)
99TryRegisterAssembly(env.ComponentCatalog, a); 223TryRegisterAssembly(env.ComponentCatalog, assembly); 289TryRegisterAssembly(_env.ComponentCatalog, args.LoadedAssembly);
ComponentModel\ComponentCatalog.cs (1)
1005LoadableClassInfo info = env.ComponentCatalog.FindClassCore(new LoadableClassInfo.Key(nameLower, signatureType));
Data\ModelLoading.cs (1)
248EnsureLoaderAssemblyIsRegistered(env.ComponentCatalog);
Data\ServerChannel.cs (1)
183if (!env.ComponentCatalog.TryFindComponent(typeof(IServerFactory), "mini", out component))
Microsoft.ML.Core.Tests (19)
UnitTests\TestEntryPoints.cs (19)
40Env.ComponentCatalog.RegisterAssembly(typeof(ExponentialAverageTransform).Assembly); 238var catalog = Env.ComponentCatalog; 266var catalog = Env.ComponentCatalog; 327Env.ComponentCatalog.RegisterAssembly(typeof(LightGbmBinaryModelParameters).Assembly); 328Env.ComponentCatalog.RegisterAssembly(typeof(TensorFlowTransformer).Assembly); 329Env.ComponentCatalog.RegisterAssembly(typeof(ImageLoadingTransformer).Assembly); 330Env.ComponentCatalog.RegisterAssembly(typeof(SymbolicSgdLogisticRegressionBinaryTrainer).Assembly); 331Env.ComponentCatalog.RegisterAssembly(typeof(SaveOnnxCommand).Assembly); 332Env.ComponentCatalog.RegisterAssembly(typeof(TimeSeriesProcessingEntryPoints).Assembly); 333Env.ComponentCatalog.RegisterAssembly(typeof(ParquetLoader).Assembly); 335var catalog = Env.ComponentCatalog; 365var catalog = Env.ComponentCatalog; 2789Env.ComponentCatalog.RegisterAssembly(typeof(LightGbmBinaryModelParameters).Assembly); 2796Env.ComponentCatalog.RegisterAssembly(typeof(LightGbmBinaryModelParameters).Assembly); 3334var catalog = Env.ComponentCatalog; 3391var catalog = Env.ComponentCatalog; 6475Env.ComponentCatalog.RegisterAssembly(typeof(TensorFlowTransformer).Assembly); 6491Env.ComponentCatalog.RegisterAssembly(typeof(TensorFlowTransformer).Assembly); 6674Env.ComponentCatalog.RegisterAssembly(typeof(LightGbmBinaryModelParameters).Assembly);
Microsoft.ML.Data (7)
Commands\ScoreCommand.cs (1)
316info = environment.ComponentCatalog.GetLoadableClassInfo<SignatureDataScorer>(loadName);
EntryPoints\EntryPointNode.cs (5)
483if (!env.ComponentCatalog.TryFindEntryPoint(entryPointName, out _entryPoint)) 489_inputBuilder = new InputBuilder(_host, _entryPoint.InputType, env.ComponentCatalog); 573bool success = env.ComponentCatalog.TryFindEntryPoint(entryPointName, out info); 576var inputBuilder = new InputBuilder(env, info.InputType, env.ComponentCatalog); 842public ComponentCatalog Catalog => _host.ComponentCatalog;
Utilities\ComponentCreation.cs (1)
294var classes = env.ComponentCatalog.FindLoadableClasses<TArgs, TSig>();
Microsoft.ML.Maml (5)
HelpCommand.cs (5)
144var infos = _env.ComponentCatalog.FindLoadableClasses(name) 195var infos = _env.ComponentCatalog.GetAllClasses() 263var sigs = _env.ComponentCatalog.GetAllSignatureTypes(); 279var infos = _env.ComponentCatalog.GetAllDerivedClasses(typeRes, typeSig) 329var sigs = _env.ComponentCatalog.GetAllSignatureTypes()
Microsoft.ML.PerformanceTests (8)
Helpers\EnvironmentFactory.cs (8)
23environment.ComponentCatalog.RegisterAssembly(typeof(TLoader).Assembly); 24environment.ComponentCatalog.RegisterAssembly(typeof(TTransformer).Assembly); 25environment.ComponentCatalog.RegisterAssembly(typeof(TTrainer).Assembly); 39environment.ComponentCatalog.RegisterAssembly(typeof(TEvaluator).Assembly); 40environment.ComponentCatalog.RegisterAssembly(typeof(TLoader).Assembly); 41environment.ComponentCatalog.RegisterAssembly(typeof(TTransformer).Assembly); 42environment.ComponentCatalog.RegisterAssembly(typeof(TTrainer).Assembly); 44environment.ComponentCatalog.RegisterAssembly(typeof(OneHotEncodingTransformer).Assembly);
Microsoft.ML.Predictor.Tests (2)
TestPredictors.cs (2)
45environment.ComponentCatalog.RegisterAssembly(typeof(LightGbmBinaryModelParameters).Assembly); 46environment.ComponentCatalog.RegisterAssembly(typeof(SymbolicSgdLogisticRegressionBinaryTrainer).Assembly);
Microsoft.ML.ResultProcessor (3)
ResultProcessor.cs (3)
158var cls = env.ComponentCatalog.GetLoadableClassInfo<SignatureTrainer>(predictorName); 525trainerClass = env.ComponentCatalog.GetLoadableClassInfo<SignatureTrainer>(commandLineTrainer.Name); 682commandClass = env.ComponentCatalog.GetLoadableClassInfo<SignatureCommand>(kind);
Microsoft.ML.Sweeper (1)
SweepResultEvaluator.cs (1)
44var evaluators = _host.ComponentCatalog.GetAllDerivedClasses(typeof(IMamlEvaluator), typeof(SignatureMamlEvaluator));
Microsoft.ML.TestFramework (11)
DataPipe\Parquet.cs (1)
16Env.ComponentCatalog.RegisterAssembly(typeof(ParquetLoader).Assembly);
DataPipe\PartitionedFileLoaderTests.cs (1)
17Env.ComponentCatalog.RegisterAssembly(typeof(ParquetLoader).Assembly);
DataPipe\TestDataPipeBase.cs (1)
350var info = env.ComponentCatalog.GetLoadableClassInfo(commandLineFactory.Name, commandLineFactory.SignatureType);
EnvironmentExtensions.cs (8)
20env.ComponentCatalog.RegisterAssembly(typeof(TextLoader).Assembly); // ML.Data 21env.ComponentCatalog.RegisterAssembly(typeof(LinearModelParameters).Assembly); // ML.StandardTrainers 22env.ComponentCatalog.RegisterAssembly(typeof(OneHotEncodingTransformer).Assembly); // ML.Transforms 23env.ComponentCatalog.RegisterAssembly(typeof(FastTreeBinaryModelParameters).Assembly); // ML.FastTree 24env.ComponentCatalog.RegisterAssembly(typeof(EnsembleModelParameters).Assembly); // ML.Ensemble 25env.ComponentCatalog.RegisterAssembly(typeof(KMeansModelParameters).Assembly); // ML.KMeansClustering 26env.ComponentCatalog.RegisterAssembly(typeof(PcaModelParameters).Assembly); // ML.PCA 27env.ComponentCatalog.RegisterAssembly(typeof(CVSplit).Assembly); // ML.EntryPoints
Microsoft.ML.Tests (2)
ExpressionLanguageTests\ExpressionLanguageTests.cs (1)
50Env.ComponentCatalog.RegisterAssembly(typeof(TestFuncs1).Assembly);
OnnxConversionTest.cs (1)
738Env.ComponentCatalog.RegisterAssembly(typeof(OnnxExportExtensions).Assembly);
Microsoft.ML.TimeSeries.Tests (1)
TimeSeries.cs (1)
20Env.ComponentCatalog.RegisterAssembly(typeof(ExponentialAverageTransform).Assembly);
Microsoft.ML.Transforms (3)
Expression\LambdaBinder.cs (1)
39_providers = env.ComponentCatalog.GetAllDerivedClasses(typeof(IFunctionProvider), typeof(SignatureFunctionProvider))
LambdaTransform.cs (2)
67env.ComponentCatalog.RegisterAssembly(assembly); 70object factoryObject = env.ComponentCatalog.GetExtensionValue(env, typeof(CustomMappingFactoryAttributeAttribute), contractName);