Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- /// <summary>
- /// The main program class.
- /// </summary>
- public class Program
- {
- // filenames for training and test data
- private static string trainingDataPath = Path.Combine(Environment.CurrentDirectory, "train_data.csv");
- private static string testDataPath = Path.Combine(Environment.CurrentDirectory, "test_data.csv");
- /// <summary>
- /// The main application entry point.
- /// </summary>
- /// <param name="args">The command line arguments.</param>
- public static void Main(string[] args)
- {
- // set up a machine learning context
- var mlContext = new MLContext();
- // set up a text loader
- var textLoader = mlContext.Data.CreateTextLoader(
- new TextLoader.Options()
- {
- Separators = new[] { ',' },
- HasHeader = true,
- AllowQuoting = true,
- Columns = new[]
- {
- new TextLoader.Column("Label", DataKind.Boolean, 1),
- new TextLoader.Column("Pclass", DataKind.Single, 2),
- new TextLoader.Column("Name", DataKind.String, 3),
- new TextLoader.Column("Sex", DataKind.String, 4),
- new TextLoader.Column("RawAge", DataKind.String, 5), // <-- not a float!
- new TextLoader.Column("SibSp", DataKind.Single, 6),
- new TextLoader.Column("Parch", DataKind.Single, 7),
- new TextLoader.Column("Ticket", DataKind.String, 8),
- new TextLoader.Column("Fare", DataKind.Single, 9),
- new TextLoader.Column("Cabin", DataKind.String, 10),
- new TextLoader.Column("Embarked", DataKind.String, 11)
- }
- }
- );
- // load training and test data
- Console.WriteLine("Loading data...");
- var trainingDataView = textLoader.Load(trainingDataPath);
- var testDataView = textLoader.Load(testDataPath);
- // the rest of the code goes here...
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement